-
Notifications
You must be signed in to change notification settings - Fork 418
Open
Description
Hi,
When I read the code, I found something hard to understand:
- In the code train.py, you randomly chose a scale index for training image every cfg.disp_interval. (https://github.com/longcw/yolo2-pytorch/blob/master/train.py#L120)
- But in the code imdb.py, you used such index for generating bounding-box information for whole images in the epoch (https://github.com/longcw/yolo2-pytorch/blob/master/datasets/imdb.py#L68). Then in every iteration in such epoch, you used the updated index for resizing training images (line 57).
I think procedure 1 and 2 are incompatible, which leads to mismatch between images and bounding-boxes. Another interesting thing is during training, more and more errors reported from darknet.py line116 occur. I think this is abnormal and may caused by 'bugs' I mentioned before. If someone has any ideas, please let me know.
Thanks a lot!
Metadata
Metadata
Assignees
Labels
No labels