train_net(config) File "train.py", line 111, in train_net for batch_idx, (imgs, gts, gt_masks, thresh_maps, thresh_masks) in enumerate(train_loader): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 681, in __next__ data = self._next_data()...
for img_path in tqdm(img_list): frame = cv2.imread(img_path) frames.append(frame) return frames class NeRFReal(BaseReal): def __init__(self, opt, trainer, data_loader, debug=True): super().__init__(opt) @@ -44,6 +54,12 @@ def __init__(self, opt, trainer, data_loader,...