search

JaryHuang / awesome_SSD_FPN_GIoU

This repository carries out some paper recurring work
106 stars 25 forks source link

How to not use FPN? #4

Open HikariTJU opened 5 years ago

HikariTJU commented 5 years ago

Default config is SSD+FPN+GIOU, right? Is there any way to train with SSD+GIOU?

JaryHuang commented 5 years ago

our network composed of three part:Backbone,Neck,and head. the Neck is FPN. Default config is SSD+FPN+GIOUS. if you want to train SSD+GIOU, you can remover the Neck.

HikariTJU commented 5 years ago

our network composed of three part:Backbone,Neck,and head. the Neck is FPN. Default config is SSD+FPN+GIOUS. if you want to train SSD+GIOU, you can remover the Neck.

So just remove x = self.neck(x) is ok?

JaryHuang commented 5 years ago

I'm sorry, 1) in build_ssd.py, make the neck is None Screenshot from 2019-07-19 20-08-04 2) in config.py you should make the backbone_out is similar neckout Screenshot from 2019-07-19 20-06-53