При выводе изображения по CSV появляется ошибка. Что не так в моем коде?

Question

[q2119999]

Обучаюсь созданию НС, пытаюсь создать СНС по лекции. Но у меня возникла проблема, когда пытаюсь вывести изображение по CSV таблице через opencv появляется ошибка. Я новичок, так что не ругайтесь сильно. Ктонибудь может объяснить в чем заключается ошибка ? Если надо могу скинуть ноут с полным кодом

Ниже часть кода

img mask
0 data\first\33-1.jpg data\mas\33-2.jpg
1 data\first\33-3.jpg data\mas\33-4.jpg
2 data\first\33-5.jpg data\mas\33-6.jpg
3 data\first\33-7.jpg data\mas\33-8.jpg
4 data\first\33-9.jpg data\mas\33-10.jpg
5 data\first\33-11.jpg data\mas\33-12.jpg
6 data\first\33-13.jpg data\mas\33-14.jpg
img, mask = train_df.iloc[6]
​
img = cv2.imread('data/first/{}'.format(img_name))
mask = cv2.imread('data/mas/{}'.format(img_name))
​
​
​
fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(25, 25))
cv2.imshow(mask[..., 0])
cv2.imshow(img)
​
plt.show()
fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(25, 25))
axes[0].imshow(img)
axes[1].imshow(mask)
plt.show()
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
in
1 fig, axes = plt.subplots(nrows=1, ncols=2, figsize=(25, 25))
2 axes[0].imshow(img)
----> 3 axes[1].imshow(mask)
4 plt.show()

~\anaconda3\envs\deeplearning\lib\site-packages\matplotlib\__init__.py in inner(ax, data, *args, **kwargs)
1445 def inner(ax, *args, data=None, **kwargs):
1446 if data is None:
-> 1447 return func(ax, *map(sanitize_sequence, args), **kwargs)
1448
1449 bound = new_sig.bind(ax, *args, **kwargs)

~\anaconda3\envs\deeplearning\lib\site-packages\matplotlib\axes\_axes.py in imshow(self, X, cmap, norm, aspect, interpolation, alpha, vmin, vmax, origin, extent, filternorm, filterrad, resample, url, **kwargs)
5521 resample=resample, **kwargs)
5522
-> 5523 im.set_data(X)
5524 im.set_alpha(alpha)
5525 if im.get_clip_path() is None:

~\anaconda3\envs\deeplearning\lib\site-packages\matplotlib\image.py in set_data(self, A)
700 if (self._A.dtype != np.uint8 and
701 not np.can_cast(self._A.dtype, float, "same_kind")):
--> 702 raise TypeError("Image data of dtype {} cannot be converted to "
703 "float".format(self._A.dtype))
704

TypeError: Image data of dtype object cannot be converted to float

Answer 1

[Vindicar]

img = cv2.imread('data/first/{}'.format(img_name))
mask = cv2.imread('data/mas/{}'.format(img_name))

Во-первых, стоит указать вторым параметром cv2.IMREAD_COLOR.
Во-вторых, проверь, что в итоге в img и mask.
В случае ошибки imread() молча возвращает None, а не кидает исключение.

Comments

[q2119999]

Параметр указал, спасибо)
У меня img это исходное изображение в jpg, а mask его маска где белым помечен объект для сегментации, остальное черным. Причем, дело явно не в них, так как по отдельности они выводятся прекрасно. А еще, если кинуть маски и исходники в одну папку, то выводятся два одинаковых изображения, т.е либо маска-маска, либо исходник-исходник. Может проблема конечно в том, что изначально по лекции маски были закодированы алгоритмом RLE и внесены во второй столбец таблицы, после чего с помощью rle_decode маска выводилась, это как я понял. Но на лекции не было объяснения как осуществить такую кодировку и в инете тоже доступной информации нет, поэтому попытался переписать код чтобы вывод шел уже из папки. Уже 3 день не могу разобраться в этом, буду благодарен если сможешь объяснить такому тупице как я что не так

[q2119999]

Так, я сумел закодировать маски, но та же проблема остается. Причем, если загрузить оригинальный датасет все ок, но дело в том, что он идентичен, единственное, что меняется, это сами изображения. Я уже и не понимаю что делать

[Vindicar]

Нужен код (можно ссылку на пастебин) с комментариями. Слёту я не пойму что происходит, что ты пытаешься сделать, и где всё идёт не так.