search

bubbliiiing / yolov4-tiny-keras

这是一个YoloV4-tiny-keras的源码，可以用于训练自己的模型。
MIT License
160 stars 58 forks source link

data_generator的anchor好像有点错误 #8

Open zezisme opened 4 years ago

zezisme commented 4 years ago

在计算anchor和true_box的iou时（train.py文件第107行开始），用到了归一化的true_boxes尺寸（在train.py的81和82行），但是anchor却没有进行归一化就与true_box进行的iou计算,从而来判断true_box属于哪个anchor，这肯定是有问题的，一个数值很大，一个只有0-1范围，从而导致anchor不能正确匹配，我是在单步调试过程中发现。不知道up主怎样认为

bubbliiiing commented 4 years ago

再看看计算iou用的到底是哪个量

zezisme commented 4 years ago

再看看计算iou用的到底是哪个量

计算iou用的就是没有归一化的anchors和归一化后的true_box呀，anchor是通过get_anchors()函数得到的，期间没有经过归一化处理，但是anchor.txt中的anchors是根据真实图片设定的anchors，如有遗漏没看到地方望up主指点

bubbliiiing commented 4 years ago

你print一下吧。 intersect_mins = np.maximum(box_mins, anchor_mins) intersect_maxes = np.minimum(box_maxes, anchor_maxes)