dlibの訓練画像を作りたいのですが、画像をxml形式で保存しなければなりません。
以下のようにopencvで読み込み,文字型に変換してから、保存しようとして見ていますが上手くできません。
何か良い方法、解決策教えていただけないでしょうか?。
python
1 2from PIL import Image 3import os 4import _pickle as cPickle 5import cv2, matplotlib 6import numpy as np 7from PIL import Image 8import xml.etree.ElementTree as ET 9 10path='/Users/Downloads/dlibtrain_img' # 画像フォルダパス 11enc_file=[] 12for i in os.listdir(path): 13 enc_file.append(base64.b64encode(str(cv2.imread(path+'/'+i)).encode("utf-8"))) 14 15for i in enc_file: 16 root = ET.fromstring(i) 17 tree = ET.ElementTree(root) 18 tree.write('train.xml') 19 20ParseError: not well-formed (invalid token): line 1, column 839 21
解決
# In[34]: enc_train_file=[] for i in train_file: enc_train_file.append(Image.open(i).convert('RGB')) # In[40]: enc_test_file=[] for i,v in enumerate(test_file): if (not i % 5 == 0): enc_test_file.append(Image.open(v).convert('RGB')) # In[41]: train_enc_img=[] for i in enc_train_file: img_bin = io.BytesIO() #メモリに保持してディレクトリ偽装みたいなことする pil_img = i.save(img_bin,"JPEG") #PILで読み込む img_bin.seek(0) byteImg=img_bin.read() data_bin=io.BytesIO(byteImg) bin_img=data_bin.getvalue() train_enc_img.append(base64.b64encode(bin_img)) for i in train_enc_img: with open('train.xml', "wb") as f: f.write(i)
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。