pytorch 从头开始faster-rcnn(三):vgg16 (带有网络冻结的写法)

由于pytorch自带了一些网络结构可以直接使用,所以直接调用有的模型,详细的可以查看官方文档:https://pytorch.org/docs/stable/torchvision/models.html def decom_vgg16(): if opt.caffe_pretrain: model = vgg16(pretrained=False) if
相关文章
相关标签/搜索