easy_detection icon indicating copy to clipboard operation
easy_detection copied to clipboard

请问inference.py应该怎么运行起来呢?麻烦解答,谢谢了!

Open Beeeee-cyber opened this issue 2 years ago • 3 comments

运行命令为:python3 inference.py --config configs/faster_rcnn_voc.yml 代码中的LOAD_CHECKPOINT、INFERENCE_LIST、IMAGE_FOLDER均改为自己的数据集和权重。 错误为:File "/home/easy_detection-master/network/Faster_RCNN/Model.py", line 40, in init if 'SCALE' in config.DATA: AttributeError: 'Option' object has no attribute 'DATA'

感谢!

Beeeee-cyber avatar Mar 09 '22 14:03 Beeeee-cyber

我也是出现相同问题,请问下原作者inference.py是预测部分吗?

ToneZe avatar Jun 12 '22 06:06 ToneZe

我也是出现相同问题,请问下原作者inference.py是预测部分吗?

我和你遇到了相同的问题,以下是我的解决方法: 首先将inference.py中48-49行代码

model = Model(opt)
model = model.to(device=opt.device)

修改为

model = Model(config)
model = model.to(device=config.device)

而后在yml文件中增加字段NUM_CLASSESS与device,例如:

MODEL:
  NAME: Cascade_RCNN
  BACKBONE: resnet50
DATA:
  DATASET: voc
  TRANSFORM: frcnn
  SCALE: [800, 1333]
  NUM_CLASSESS: 2
OPTIMIZE:
  OPTIMIZER: sgd
  BASE_LR: 0.001 
  SCHEDULER: mini 
  BATCH_SIZE: 1
TEST:
  NMS_THRESH : 0.5  # nms iou thresh at test time
  CONF_THRESH: 0.05 # confidence thresh to keep at test time
MISC:
  VAL_FREQ: 1
  SAVE_FREQ: 1
  NUM_WORKERS: 0
device: cuda:0

希望能对你有所帮助

ddddddqw avatar Mar 05 '23 15:03 ddddddqw

哥们,真的非常 感谢,按照你的方法进行修改,真的运行起来了,真的谢谢了!!!!@ddddddqw

yuanfangshang888 avatar Nov 19 '23 08:11 yuanfangshang888