I am training an object detection model on my Dataset using Faster RCNN and resnet50 with pre-trained weights. I am primarily following the following link while implementing it, the only difference is that instead of segmentation, I am performing object de...

我正在尝试将5维张量（从自定义特征提取器提取特征后）作为输入输入到Faster RCNN网络，以训练对象检测模型。但是，我在标准化输入时遇到错误。我的代码如下 num_classes = 5 model = fasterrcnn_resnet50_fpn(pretrained=False) in_features = model.roi_heads.box_predictor.cls_score.in_features model.backbone.body.conv1 = Conv2d(5, 64, ke...