u2netdemo icon indicating copy to clipboard operation
u2netdemo copied to clipboard

ValueError: Could not find a format to read the specified file in single-image mode

Open matrs opened this issue 2 years ago • 0 comments

Hey, thank you for your effort in sharing this. I've tried all the collab notebooks and all are failling

# run the test script, and outputs are saved to results folder
!python -W ignore u2net_test.py
['/content/U-2-Net/images/2000388537240_2.png', '/content/U-2-Net/images/2000388299407_2.png']
Traceback (most recent call last):
  File "u2net_test.py", line 117, in <module>
    main()
  File "u2net_test.py", line 91, in main
    for i_test, data_test in enumerate(test_salobj_dataloader):
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 681, in __next__
    data = self._next_data()
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1376, in _next_data
    return self._process_data(data)
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1402, in _process_data
    data.reraise()
  File "/usr/local/lib/python3.7/dist-packages/torch/_utils.py", line 461, in reraise
    raise exception
ValueError: Caught ValueError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 302, in _worker_loop
    data = fetcher.fetch(index)
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/content/U-2-Net/data_loader.py", line 242, in __getitem__
    image = io.imread(self.image_name_list[idx])
  File "/usr/local/lib/python3.7/dist-packages/skimage/io/_io.py", line 48, in imread
    img = call_plugin('imread', fname, plugin=plugin, **plugin_args)
  File "/usr/local/lib/python3.7/dist-packages/skimage/io/manage_plugins.py", line 207, in call_plugin
    return func(*args, **kwargs)
  File "/usr/local/lib/python3.7/dist-packages/skimage/io/_plugins/imageio_plugin.py", line 10, in imread
    return np.asarray(imageio_imread(*args, **kwargs))
  File "/usr/local/lib/python3.7/dist-packages/imageio/core/functions.py", line 265, in imread
    reader = read(uri, format, "i", **kwargs)
  File "/usr/local/lib/python3.7/dist-packages/imageio/core/functions.py", line 182, in get_reader
    "Could not find a format to read the specified file in %s mode" % modename
ValueError: Could not find a format to read the specified file in single-image mode

Then, the next cells fail as expected

matrs avatar Sep 05 '22 17:09 matrs