Question

我有一个仅与colab.upload一起使用的功能，而不是通过我从云端硬盘上传的colab磁盘上文件中的文件名传递给它的。我想为TensorFlow对象检测API使用jpg图片。

这有效：

uploaded = files.upload()
img_infer = list(uploaded)[0]

print('running test on: ' + img_infer)
img_inference(img_infer)

但不是这样：

for file in os.listdir("/content/test"):
    print('running test on: ' + file)
    img_infer = list(file)[0]
    img_inference(img_infer)

错误：

---------------------------------------------------------------------------

FileNotFoundError                         Traceback (most recent call last)

<ipython-input-31-4fbcedf7108a> in <module>()
     12   print('Teste mit Bild: ' + file)
     13   img_infer = list(file)[0]
---> 14   img_inference(img_infer)

2 frames

<ipython-input-8-6a0800ccaf6f> in img_inference(img_path)
     50 
     51 def img_inference(img_path):
---> 52   image = read_image_bgr(img_infer)
     53 
     54   # copy to draw on

/content/keras-retinanet/keras_retinanet/utils/image.py in read_image_bgr(path)
     30     """
     31     # We deliberately don't use cv2.imread here, since it gives no feedback on errors while reading the image.
---> 32     image = np.asarray(Image.open(path).convert('RGB'))
     33     return image[:, :, ::-1].copy()
     34 

/usr/local/lib/python3.6/dist-packages/PIL/Image.py in open(fp, mode)
   2528 
   2529     if filename:
-> 2530         fp = builtins.open(filename, "rb")
   2531         exclusive_fp = True
   2532 

FileNotFoundError: [Errno 2] No such file or directory: 'I'

Answer 1

您的代码中存在逻辑问题：

for file in os.listdir("/content/test"):
  print('running test on: ' + file)
  img_infer = list(file)[0] # <== This line is not correct
  img_inference(img_infer)

您必须将img_infer = list(file)[0]更改为img_infer = file，或者仅删除img_infer = list(file)[0]并将img_inference(img_infer)更改为img_inference(file)。

换句话说：

for file in os.listdir("/content/test"):
  print('running test on: ' + file)
  img_inference(file)

为进一步说明原始代码中发生的情况，假设/content/test包含aaa.txt，bbb.py，ccc.jpg。然后：

for file in os.listdir("/content/test"):
  # In the first iteration file == "aaa.txt"
  print('running test on: ' + file)
  # But then list(file) == ["a", "a", "a", ".", "t", "x", "t"] which is not what you want.
  # And then list(file)[0] == "a", therefore img_infer == "a"
  img_infer = list(file)[0]
  img_inference(img_infer)

希望有帮助！

Google Colab从colabdisk读取文件而不是上传

1 个答案: