Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
kkkkf666 opened this issue Jan 12, 2022 · 2 comments
Open

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

kkkkf666 opened this issue Jan 12, 2022 · 2 comments

Comments

@kkkkf666
Copy link

No description provided.

@yangxue0827
Copy link
Member

用p2-p6得特征层试一下
@kkkkf666
Copy link
Author

@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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
2 participants
@yangxue0827 @kkkkf666