我正在使用 TensorFlow 和 Python 制作图像分类器,但使用 CV2 读取图像时出错。我对 CV2 很陌生,我还没有找到任何可以充分解决我的问题的东西。谁能解释一下如何解决这个问题?
def train_data_with_label():
train_images = []
for i in tqdm(os.listdir(train_data)):
path = os.path.join(train_data, i)
img = cv2.imread(path, 3)
img = cv2.resize(img, (64,64))
train_images.append([np.array(img), one_hot_label(i)])
shuffle(train_images)
return train_images
def test_data_with_label():
test_images = []
for i in tqdm(os.listdir(test_data)):
path = os.path.join(test_data, i)
img = cv2.imread(path, 3)
img = cv2.resize(img, (64,64))
test_images.append([np.array(img), one_hot_label(i)])
shuffle(test_images)
return test_images
Run Code Online (Sandbox Code Playgroud)
这是我得到的错误:
Using TensorFlow backend.
0%| | 0/2 [00:00<?, ?it/s]
Traceback (most recent call last):
File "retrain.py", line 47, in …Run Code Online (Sandbox Code Playgroud)