import torch
import torchvision.models as models
import torchvision
net = models.detection.fasterrcnn_resnet50_fpn(pretrained=False)
pthfile = r'G:\Anaconda3\envs\pytorch\Lib\site-packages\torchvision\models\fasterrcnn_resnet50_fpn_coco-258fb6c6.pth'
net.load_state_dict(torch.load(pthfile))
images, boxes = torch.rand(4, 3, 600, 1200), torch.rand(4,
pytorch读取VOC数据之VOCDetection之二
最新推荐文章于 2024-05-19 22:54:49 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)