使用pytorch搭建faster RCNN的代码

下面是一个基本的 Faster R-CNN 模型搭建代码：

import torch
import torchvision# 定义模型
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)

# 定义优化器和损失函数
optimizer = torch.optim.SGD(model.parameters(), lr=0.005, momentum=0.9)
loss_func = torch.nn.CrossEntropyLoss()

# 训练模型
for epo