u2netdemo
u2netdemo copied to clipboard
ValueError: Could not find a format to read the specified file in single-image mode
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