search

naseemap47 / YOLO-NAS

Train and Inference your custom YOLO-NAS model by Single Command Line
Apache License 2.0
98 stars 13 forks source link

Batch #37

Closed naseemap47 closed 1 year ago

naseemap47 commented 1 year ago

📺 Batch Inference

Arguments: -n, --num: Number of classes the model trained on -m, --model: Model type (choices: yolo_nas_s, yolo_nas_m, yolo_nas_l) -w, --weight: path to trained model weight -s, --source: paths to videos/cam-ids/RTSPs --full: to enable full screen

python3 batch.py --num 3 --model yolo_nas_m --weight /runs/train4/ckpt_best.pth --source '/test/video.mp4' '/test/video23.mp4'      # videos
                                                                                --source 0 2 --conf 0.45 --full                     # web-Cameras with full screen
                                                                                --source 'rtsp://link' 'rtsp://link3' --conf 0.25   # RTSPs video stream