Error : Could not read image in Google Colab

Error : Could not read image in Google Colab

我正在按照本教程在 Detect 上构建自定义对象检测模型。

https://www.analyticsvidhya.com/blog/2021/06/simplest-way-to-do-object-detection-on-custom-datasets/

我已经收集并标记了我的图像,将它们放在我的云端硬盘上,我正在 运行 以下代码片段来训练模型,该模型是 Python Notebook 的一部分 Google Colab:

Train_dataset = core.Dataset('/content/drive/My Drive/training model/Training',transform=custom_transforms)#L1
Test_dataset = core.Dataset('/content/drive/My Drive/training model/Test')#L2

loader=core.DataLoader(Train_dataset, batch_size=2, shuffle=True)#L3

model = core.Model(['black car', 'grey car','white truck'])#L4

losses = model.fit(loader, Test_dataset, epochs=25, lr_step_size=5, learning_rate=0.001, verbose=True)#L5
plt.plot(losses)
plt.show()

但是,在第一个模型时期开始后不久,我不断收到以下错误:

ValueError: Could not read image /content/drive/My Drive/training model/Training/frame22.jpg

它随机给出这个错误,不仅是 frame22,还有其他不存在于该目录中的框架。我试图通过在脚本开头启用 force_remount 来重新安装我的驱动器,但错误仍然存​​在。

我检查了 Detecto 的 core.Dataset 实现代码,并确认了我在评论中所说的内容。

索引是通过获取所有 .xml 注释文件并创建将它们映射到图像的索引来创建的。它不会检查图像是否确实存在。

对于图像文件名,它使用 inside xml 文件的名称,而不是 xml 文件的名称。请参阅下面的注释 XML 文件视图,您可以在其中看到 filename 属性。如果更改图像的名称,则需要在 xml 文件中进行更改。