我正在尝试训练从Detectron2获取的对象检测网络,并将其加载到Pytorch中。它必须检测苹果。问题是,当我使用forward()函数时,在功能中会出现以下错误:
train_one_epoch(py_model,优化器,data_loader,设备,纪元,args.print_freq)
在train_one_epoch的第30行中,文件“ /home/felipevw/MyAppleDetector/utility/engine.py”
loss_dict =模型(图像,目标)
在调用中,文件“ /home/felipevw/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py”,第532行
结果= self.forward(* input,** kwargs)
TypeError:forward()接受2个位置参数,但给出了3个
如果有人可以帮助我,我将不胜感激。我的代码是:
import datetime
import os
import time
import torch
import torch.utils.data
import torchvision
from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
from torchvision.models.detection.mask_rcnn import MaskRCNNPredictor
fr