使用 cv2 收集图像数据时出现问题

Question

我正在尝试使用 cv2 从一些猫和狗图像中获取图像数据，用于 python 和 append 中的机器学习项目，将它们全部添加到training_data列表中。 但它只是在最后打印列表时打印None 。

起初我想也许我需要将数据转换为 numpy 数组，但似乎没有任何效果，我真的不明白出了什么问题。

CATEGORIES = ["Dog", "Cat"]     # 0=dog, 1=cat
IMG_SIZE = 50

training_data = []

def create_training_data():
    i = 0
    for category in CATEGORIES:
        path = os.path.join(DATADIR, category)  # path to cats or dogs dir
        class_num = CATEGORIES.index(category)
        for img in os.listdir(path):
            i += 1
            try:
                img_array = cv2.imread(os.path.join(path, img), cv2.IMREAD_GRAYSCALE)
                new_array = cv2.resize(img_array, (IMG_SIZE, IMG_SIZE))
                training_data.append([new_array, class_num])
            except Exception as e:
                print("general exception", e, os.path.join(path, img))
            if i % 1000 == 0:
                print(i, '...')
    print('\n\nfinished...\n\n')

create_training_data()

training_data = random.shuffle(training_data)

print(training_data)

X = []
y = []

for features, label in training_data:
    X.append(features)
    y.append(label)

X = np.array(X).reshape(-1, IMG_SIZE, IMG_SIZE, 1)

我认为这与调整大小有关，因为它引发了很多这样的异常：

一般异常 OpenCV(4.1.1) C:...\opencv\modules\imgproc\src\resize.cpp:3720: error: (-215:Assertion failed).ssize:empty() in function:resizecv:resize ' D..../cats_and_dogs/PetImages\Cat\9565.jpg

它还打印此错误：

回溯（最近一次通话最后）：

文件“D:/Python/tensorflow/tutorial/cats_and_dogs.py”，第 44 行，在功能中，label 在 training_data：TypeError：'NoneType' object 不可迭代

损坏的 JPEG 数据：标记 0xd9 之前的 399 个无关字节

损坏的 JPEG 数据：标记 0xd9 之前的 226 个无关字节

损坏的 JPEG 数据：标记 0xd9 之前的 162 个无关字节

警告：未知的 JFIF 修订号 0.00

...

Answer 1

在研究了您的错误后，我发现这个线程可能与您遇到的问题有关。

简而言之，该线程指出了 Oxford-IIIT 宠物数据集的问题。 在该数据集中，一些图像以“.jpg”扩展名结尾，但实际上是“.png”。 验证您从哪里获得图像。

看来您遇到了类似/相同的问题。 文件扩展名与实际图像数据不匹配可能导致 openCV 无法正确读取数据。

编辑：2019 年 10 月 14 日

此外， random.shuffle()就地运行，因此返回None 。 将training_data = random.shuffle(training_data)更改为random.shuffle(training_data)