easy_detection
easy_detection copied to clipboard
请问inference.py应该怎么运行起来呢?麻烦解答,谢谢了!
运行命令为: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'
感谢!
我也是出现相同问题,请问下原作者inference.py是预测部分吗?
我也是出现相同问题,请问下原作者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