FasterRCNN-pytorch icon indicating copy to clipboard operation
FasterRCNN-pytorch copied to clipboard

权重在哪下载

Open huangshaoguang opened this issue 5 years ago • 2 comments

你好!非常感谢你的分享,请问预训练的权重在哪下载呢

huangshaoguang avatar Mar 24 '19 01:03 huangshaoguang

同问,谢谢。

lao19881213 avatar Mar 02 '20 04:03 lao19881213

@lao19881213 非常抱歉,之前的权重是自己训练的,已经找不回了。

  1. 建议使用pytorch提供权重:https://pytorch.org/docs/0.3.0/_modules/torchvision/models/vgg.html#vgg16
  2. 权重使用时注意mean和std操作:https://pytorch.org/docs/0.3.0/torchvision/models.html 以及需要在以下函数汇总修改数据图片读取格式(BGR-》RGB,并normalize it):https://github.com/yingxingde/FasterRCNN-pytorch/blob/8ea2f9899bffeb8866dc53cd00d4812513588160/lib/roi_data_layer/minibatch.py#L58
  3. 如果权重加载过程中,出现weight key不match的情况,需要手动在此处修正:https://github.com/yingxingde/FasterRCNN-pytorch/blob/master/lib/nets/vgg16.py

希望使用愉快~

yingxingde avatar Jun 09 '20 11:06 yingxingde