search

YuHengsss / YOLOV

This repo is an implementation of PyTorch version YOLOV Series
Apache License 2.0
309 stars 43 forks source link

TypeError: YOLOX.forward() got an unexpected keyword argument 'nms_thresh' #50

Closed timmermansjoy closed 1 year ago

timmermansjoy commented 1 year ago

when trying to do the command in the readme python tools/vid_demo.py -f ./exps/yolov/yoloxs_vid.py -c ./yoloxs_vid.pth --path ./189_4_1_20230217042816000.tmp.mp4 --conf 0.25 --nms 0.5 --tsize 600 I get the following error: 

File "/Users/joy/jtest/YOLOV/tools/vid_demo.py", line 303, in <module>
    main(exp, args)
  File "/Users/joy/jtest/YOLOV/tools/vid_demo.py", line 296, in main
    imageflow_demo(predictor, vis_folder, current_time, args)
  File "/Users/joy/jtest/YOLOV/tools/vid_demo.py", line 224, in imageflow_demo
    outputs.extend(predictor.inference(ele))
  File "/Users/joy/jtest/YOLOV/tools/vid_demo.py", line 137, in inference
    outputs,outputs_ori = self.model(img, nms_thresh=self.nmsthre)
  File "/Users/joy/jtest/YOLOV/.venv/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
TypeError: YOLOX.forward() got an unexpected keyword argument 'nms_thresh'
YuHengsss commented 1 year ago

Hello, sorry for my unclear instruction, you should use demo.py for yolox model and vid_demo.py for yolov model.