foolwood / foolwood/SiamMask

siam_mask_dataset error

Open
#168 5 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
3.6k
Forks
804
PR merge metrics
No merged PRs in 30d

Description

Traceback (most recent call last):
File "/home/soyeon/SiamMask/tools/train_siammask.py", line 292, in
main()
File "/home/soyeon/SiamMask/tools/train_siammask.py", line 168, in main
train(train_loader, dist_model, optimizer, lr_scheduler, args.start_epoch, cfg)
File "/home/soyeon/SiamMask/tools/train_siammask.py", line 186, in train
for iter, input in enumerate(train_loader):
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 819, in __next__
return self._process_data(data)
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 846, in _process_data
data.reraise()
File "/opt/conda/lib/python3.6/site-packages/torch/_utils.py", line 385, in reraise
raise self.exc_type(msg)
TypeError: Caught TypeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/soyeon/SiamMask/datasets/siam_mask_dataset.py", line 551, in __getitem__
search_mask = (cv2.imread(search[2], 0) > 0).astype(np.float32)
TypeError: '>' not supported between instances of 'NoneType' and 'int'

How can I fix this error? Please...
I want to use training set without ytb. (I use only coco, vid, det)

Contributor guide

No contributing guide indexed for this repository

Research direction

Read datasets/siam_mask_dataset.py around line 551 and tools/train_siammask.py around lines 168-186, then trace which COCO, VID, or DET dataset entry produces the failing image path. Reproduce the training run without YTB and verify that the DataLoader no longer raises this TypeError.

Written by the indexing model from the issue text.

Assessment

Tech stack
numpy, opencv, python, pytorch
Domain
computer-vision, machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.