search

hhk7734 / tensorflow-yolov4

YOLOv4 Implemented in Tensorflow 2.
MIT License
136 stars 75 forks source link

Training interrupted due to index out of bounds error #2

Closed mon28 closed 4 years ago

mon28 commented 4 years ago

Hi,

My training began successfully but got interrupted with the following error after some 30 steps.

Traceback (most recent call last):
  File "train.py", line 10, in <module>
    pre_trained_weights=None,
  File "/data/mtare/darknet-py/venv/lib/python3.6/site-packages/yolov4/tf/__init__.py", line 286, in train
    for image_data, target in trainset:
  File "/data/mtare/darknet-py/venv/lib/python3.6/site-packages/yolov4/core/dataset.py", line 137, in __next__
    ) = self.preprocess_true_boxes(bboxes)
  File "/data/mtare/darknet-py/venv/lib/python3.6/site-packages/yolov4/core/dataset.py", line 373, in preprocess_true_boxes
    label[best_detect][yind, xind, best_anchor, :] = 0
IndexError: index 76 is out of bounds for axis 0 with size 76

I don't think my data is inaccurate. Let me know what you think. @hhk7734

hhk7734 commented 4 years ago

I'm doing other work, so I can't test it now. Maybe, I'm going to see it this weekend or next week.

hhk7734 commented 4 years ago

There seems to be no way for me to test this problem.