Почему этот код не может прочитать список изображений внутри папки. Я хотел прочитать изображения .jpg в тестовой папке. когда я передаю путь к тестовой папке в качестве аргумента через командную строку
# Dataset and Loader
dataset_test = TestDataset(
cfg.list_test,
cfg.DATASET)
loader_test = torch.utils.data.DataLoader(
dataset_test,
batch_size=cfg.TEST.batch_size,
shuffle=False,
collate_fn=user_scattered_collate,
num_workers=2,
drop_last=True)
# generate testing image list
if os.path.isdir(args.imgs[0]):
imgs = find_recursive(args.imgs[0])
else:
imgs = [args.imgs]
assert len(imgs), "imgs should be a path to image (.jpg) or directory."
cfg.list_test = [{'fpath_img': x} for x in imgs]
выдает следующую ошибку
IsADirectoryError: [Errno 21] Is a directory: