关于论文中红绿灯检测的是如何训练的呢？我使用Retinanet作为 base model训练出来的效果并不理想

[kkkkf666]

关于论文中红绿灯检测的是如何训练的呢？我使用Retinanet作为 base model训练出来的效果并不理想

[yangxue0827]

用p2-p6得特征层试一下

[kkkkf666]

@yangxue0827 不好意思我说错了，我现在是使用resnet50_coco_best_v2.1.0.h5作为base model，训练代码是kera-Retinanet那套进行训练，因为当时看readme中提供的是DOTA训练出来的R3Det*，预训练数据是遥感的并不是红绿灯的就没有使用，可以方便说一下要是用您这个训练红绿灯的应该怎么操作，是按照readme中操作进行吗？但是没有看到retinanet的Pretrain weights @yangxue0827 感谢您的回复

---

(1) Modify parameters (such as CLASS_NUM, DATASET_NAME, VERSION, etc.) in $PATH_ROOT/libs/configs/cfgs.py
(2) Add category information in $PATH_ROOT/libs/label_name_dict/label_dict.py
(3) Add data_name to $PATH_ROOT/data/io/read_tfrecord_multi_gpu.py