训练阶段 import torch import torchvision from torchvision.models.detection import FasterRCNN_ResNet50_FPN_Weights model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights=FasterRCNN_ResNet50_FPN_Weights.DEFAULT