通过名称和元数据加载原始图像

时间:2019-03-04 21:07:42

标签: python image csv numpy image-processing

我是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文件,该文件具有描述该图像的坐标,我想以某种方式将其与数据集结合,或者直接在训练阶段。

预先感谢

0 个答案:

没有答案