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
相關文章
相關標籤/搜索