From efc16f39283d9218514e8b507d69f4420fae5436 Mon Sep 17 00:00:00 2001 From: NASEEM A P Date: Fri, 4 Aug 2023 15:44:00 +0530 Subject: [PATCH] updated README for inference batching --- README.md | 24 ++++++++++++++++++++---- 1 file changed, 20 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 2be514c..19fbe26 100644 --- a/README.md +++ b/README.md @@ -164,8 +164,24 @@ python3 inference.py --num 3 --model yolo_nas_m --weight /runs/train4/ckpt_best. ``` ## 📺 Inference Batching -### Upcomming... -**Example:** -``` -python3 batch.py +#### Support +- Video +- Camera +- RTSP + +
+ Args + + `-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 +``` \ No newline at end of file