FasterRCNN-pytorch
FasterRCNN-pytorch copied to clipboard
权重在哪下载
你好!非常感谢你的分享,请问预训练的权重在哪下载呢
同问,谢谢。
@lao19881213 非常抱歉,之前的权重是自己训练的,已经找不回了。
- 建议使用pytorch提供权重:https://pytorch.org/docs/0.3.0/_modules/torchvision/models/vgg.html#vgg16
- 权重使用时注意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
- 如果权重加载过程中,出现weight key不match的情况,需要手动在此处修正:https://github.com/yingxingde/FasterRCNN-pytorch/blob/master/lib/nets/vgg16.py
希望使用愉快~