iamhankai / ghostnet.pytorch

[CVPR2020] GhostNet: More Features from Cheap Operations
https://arxiv.org/abs/1911.11907
522 stars 118 forks source link

用ghostnet做anchorfree目标检测的主干网络 #35

Open gp12345678 opened 4 years ago

gp12345678 commented 4 years ago

你好我用ghostnet做anchorfree目标检测的主干网络,检测七类,选取5,10,16层作为FPN输入,训练数据1.3万张。目前训练270epoch,loss=0.78,而用mnasnet做主干,100多epoch的时候loss就降到小数点后两位了。请问我这个超参如何设置合适,目前sgd、lr=0.0001、momentum=0.9、weight_decay=le-4、cosineLR。请问你可以给我点意见吗,谢谢。

gp12345678 commented 4 years ago

set_random_seed(1)

gp12345678 commented 4 years ago

bachsize=16

iamhankai commented 4 years ago

我们选了 [3,5,11,16] 这些block作为FPN的输入。超参设置和你训mnasnet一样就行吧。

gp12345678 commented 4 years ago

@iamhankai 你好,GhostNet做segmentation的backbone效果你们试过吗,我想用ghostnet做主干后面接objection和segmentation。不知道效果会怎样?

iamhankai commented 4 years ago

detection做过,还不错。