我是python和数据分析的新手,希望能获得帮助。
我正在尝试加载每个名称为ImXXX_Y.jpg的图像集,其中Y是每个图像的标签。
我尝试了以下代码:
import csv
with open('foo.csv', 'wb') as csvfile:
filewriter = csv.writer(csvfile, delimiter=',',
quotechar='|', quoting=csv.QUOTE_MINIMAL)
filewriter.writerow(['img', 'label'])
for img in os.listdir(path):
label = 0
if img.endswith('1.jpg') : label = 1
impath = img_dir +img
img = cv2.imread(impath,1)
filewriter.writerow([np.array(img),label])
foo=open('foo.csv')
print(foo)
我得到这个错误:
<_io.TextIOWrapper name='foo.csv' mode='r' encoding='cp65001'>
每个图像还具有ImXXX_Y.xyc文件,该文件具有描述该图像的坐标,我想以某种方式将其与数据集结合,或者直接在训练阶段。
预先感谢