训练时候会进去死循环，无法训练

[futureflsl]

这个项目存在严重BUG，这个我看到了#15楼一样反馈，再dataloader.py的106行，我打印输出，死循环了，希望作者修复一下 img, y = self.get_random_data(lines[index], self.image_size[0:2]) if len(y)==0: print("--------------------------") index = (index + 1) % n continue

[bubbliiiing]

你是不是很多图片没有目标

[futureflsl]

你是不是很多图片没有目标 都有目标，我把数据集用过你的其他项目都是可以的，这是你的项目包含源码+数据集，修改do_configuration.py中的root_dir就可以跑起来 链接: https://pan.baidu.com/s/1iDYHGEkWp5XwF_HF90EeTg 提取码: kby2

[bubbliiiing]

这是我刚跑的结果

[bubbliiiing]

Epoch 1/50: 100%|████████████████████████████████████████████████████████████████████████████████| 75/75 [00:17<00:00, 4.20it/s, conf_loss=9.33, loc_loss=2.95, lr=0.0005] Saving state, iter: 1

[bubbliiiing]

再另一个issue你给我的数据集300张

[futureflsl]

再另一个issue你给我的数据集300张

好吧，我再看看

[bubbliiiing]

你重下一下代码，你应该看看2007_train.txt里面是不是有目标信息，你可能voc_annotation.py的classes没改

[bubbliiiing]

我现在就是刚下的代码，刚下的数据集，直接改好参数训练的

[futureflsl]

谢谢大佬，我知道了。你其他项目生成txt格式都是个文件名，这个刚好放了box，我没放box。

[bubbliiiing]

好的