我想减小对象检测模型的大小。同样,我尝试使用pytorch mobile Optimizer优化更快的R-CNN模型进行目标检测,但生成的.pt
{
我使用了下面提到的代码
import torch
import torchvision
from torch.utils.mobile_optimizer import optimize_for_mobile
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
model.eval()
script_model = torch.jit.script(model)
from torch.utils.mobile_optimizer import optimize_for_mobile
script_model_vulkan = optimize_for_mobile(script_model, backend='Vulkan')
torch.jit.save(script_model_vulkan, "frcnn.pth")
您必须首先量化您的模型
遵循以下步骤here
&;然后使用这些方法
编辑:
resnet50模型的量化处理
相关问题 更多 >
编程相关推荐