YunYang1994 / tensorflow-yolov3

🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"
https://yunyang1994.gitee.io/2018/12/28/YOLOv3-算法的一点理解/
MIT License
3.63k stars 1.35k forks source link

about dataset.py #34

Open lxztju opened 5 years ago

lxztju commented 5 years ago

when I trained the pascal_voc2012, I got this error in dataset.py line 107, in preprocess_true_boxes y_true[l][j, i, k, 5+c] = 1.

IndexError: index 425 is out of bounds for axis 3 with size 25 Why I can got this error? If anyone has a solution ,please tell me

YunYang1994 commented 5 years ago
# image_path, x_min, y_min, x_max, y_max, category_id,  x_min, y_min, ... category_id

set debug = True in Parser, and check your feed data again,

lxztju commented 5 years ago

I set debug = True in Parser,and got image and gt_box is normal. the image is [416,416,3] and gt_box is[?,5]. I use the dataset of VOC2012, In train.py, what I need is to change the path of tfrecords and anchor s could you teach me how to train VOC dataset. thank you