Skip to content

Evaluation - masks are only present for odd frames #7

@Mona-ShZeinoddin

Description

@Mona-ShZeinoddin

Hello and thank you again for your amazing work:

When I run inference on a sequence of the StereoMIS dataset:

python infer_trajectory.py /root/mounted_point/StereoMIS/P2_0

"Stereo Dataset
0%| | 0/13825 [00:00<?, ?it/s][ WARN:0@5.369] global loadsave.cpp:268 findDecoder imread_('/root/mounted_point/StereoMIS/P2_0/masks/000002l.png'): can't open/read file: check file path/integrity
0%| | 1/13825 [00:01<6:04:05, 1.58s/it]
Traceback (most recent call last):
File "infer_trajectory.py", line 210, in
main(args, config)
File "infer_trajectory.py", line 71, in main
for i, data in enumerate(tqdm(loader, total=min(len(dataset), (args.stop-args.start)//args.step))):
File "/root/yes/envs/stmis/lib/python3.8/site-packages/tqdm/std.py", line 1181, in iter
for obj in iterable:
File "/root/yes/envs/stmis/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 628, in next
data = self._next_data()
File "/root/yes/envs/stmis/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1333, in _next_data
return self._process_data(data)
File "/root/yes/envs/stmis/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1359, in _process_data
data.reraise()
File "/root/yes/envs/stmis/lib/python3.8/site-packages/torch/_utils.py", line 543, in reraise
raise exception
cv2.error: Caught error in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/yes/envs/stmis/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 302, in _worker_loop
data = fetcher.fetch(index)
File "/root/yes/envs/stmis/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 58, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/yes/envs/stmis/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 58, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/robust-pose-estimator/scripts/../dataset/stereo_dataset.py", line 31, in getitem
mask = cv2.resize(cv2.imread(self.imgs[item].replace('video_frames', 'masks'), cv2.IMREAD_GRAYSCALE), dsize=(img_l.shape[1], img_l.shape[0]), interpolation=cv2.INTER_NEAREST)
cv2.error: OpenCV(4.11.0) /io/opencv/modules/imgproc/src/resize.cpp:4208: error: (-215:Assertion failed) !ssize.empty() in function 'resize'"

Which is due to the missing mask for the frame being loaded. I have checked and in the masks directory, masks are only present for odd frames "000001l.png, 000003l.png, 000005l.png, .....". How have you been able to run the code with the current script for mask definition? Would be very grateful for your insight.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions