索引错误:列出索引超出范围 - 打印图像时

Question

编写了一个函数来显示来自 2 个不同文件夹的 6 张胸部 X 光图像。图像被分成 2 个文件夹，包含在 1 个文件夹中。(训练数据是主文件夹，其中有 2 个文件夹名称为"PNEUMONUA"和"NORMAL"。

当我在一个文件夹中的图片上运行该功能时，它可以完美运行，当我在另一个文件夹上使用该功能时，我得到"索引错误：列出索引超出范围"。

法典：

TRAINING_DATA = "/home/jack/Desktop/chest_xray/train/"
TEST_DATA = "/home/jack/Desktop/chest_xray/test/"
VALIDATION_DATA = "/home/jack/Desktop/chest_xray/val/"

def plot_images(path, labeled=True, max_images=6):
amount = 0
fig = plt.figure(figsize=(10, 6))
for file in os.listdir(path):
if file.endswith('.jpeg'):
if amount == max_images:
break
img = mpimg.imread(os.path.join(path, file))
plt.subplot(231+amount)
if labeled:
plt.title(file.split('_')[1])
imgplot = plt.imshow(img)
amount += 1

plot_images(TRAINING_DATA + '/NORMAL')
#ERROR
plot_images(TRAINING_DATA + '/PNEUMONIA')
#WORKS FINE

有什么想法吗？

Answer 1

"NORMAL"文件夹中图像的文件名没有要拆分的"_"(例如。IM-0115-0001.jpeg，NORMAL2-IM-0666-0001.jpeg(。因此，您不能根据"_"拆分它。

我刚刚在您的代码中注释了 2 行，它工作正常。

TRAINING_DATA = "<path>/chest_xray/train/" 

def plot_images(path, labeled=True, max_images=6):
amount = 0
fig = plt.figure(figsize=(10, 6))
for file in os.listdir(path):
if file.endswith('.jpeg'):
if amount == max_images:
break
img = cv2.imread(os.path.join(path, file))
plt.subplot(231+amount)
plt.title("Normal")
imgplot = plt.imshow(img)
amount += 1
plot_images(TRAINING_DATA + '/NORMAL')

希望这有帮助。