ppyolov2训练出模型后,利用测试集进行批量推理报错
Traceback (most recent call last):
File "tools/infer.py", line 158, in <module>
main()
File "tools/infer.py", line 154, in main
run(FLAGS, cfg)
File "tools/infer.py", line 128, in run
trainer.predict(
File "D:\Anaconda\envs\paddle20\lib\site-packages\paddledet-2.0.1-py3.8.egg\ppdet\engine\trainer.py", line 399, in predict
int(outs['im_id']), catid2name,
TypeError: only size-1 arrays can be converted to Python scalars
请各位帮帮忙解决