Блог

Это исходный код тестирования модели на изображениях:

 filelist = sorted(glob.glob(test_img_folder "*.png"))  for path in filelist:  base = osp.splitext(osp.basename(path))[0]   img = cv2.imread(path, cv2.IMREAD_COLOR)  img = img * 1.0 / 255  img = torch.from_numpy(np.transpose(img[:, :, [2, 1, 0]], (2, 0, 1))).float()  img_LR = img.unsqueeze(0)  img_LR = img_LR.to(device)   with torch.no_grad():  output = model(img_LR).data.squeeze().float().cpu().clamp_(0, 1).numpy()  output = np.transpose(output[[2, 1, 0], :, :], (1, 2, 0))  output = (output * 255.0).round()  cv2.imwrite('results/{:s}_rlt.png'.format(base), output)  

Затем я пытаюсь использовать пакетную обработку для ускорения процесса. Измененный код

 filelist = sorted(glob.glob(test_img_folder "*.png")) batch_data = DataLoader(filelist,5,shuffle=False,drop_last=True)  for path in batch_data:  base = osp.splitext(osp.basename(path))[0]  ...   cv2.imwrite('results/batch/{:s}_rlt.png'.format(base), output)  

Тогда есть ошибка:

Файл «test_batch.py», строка 37, в базе = osp.splitext(osp.базовое имя(путь))[0]

Файл «/home/anaconda3/envs/torch/lib/python3.7/posixpath.py», строка 146, в базовом имени p = os.Ошибка типа fspath(p): ожидаемый str, байт или ос.Объект, подобный пути, а не список

Похоже, что загрузчик данных превращает список файлов в тип списка. Тогда как я могу получить каждое имя файла в состоянии пакетного процесса?