Skip to content

Commit

Permalink
updated README for inference batching
Browse files Browse the repository at this point in the history
  • Loading branch information
naseemap47 committed Aug 4, 2023
1 parent b5ed024 commit efc16f3
Showing 1 changed file with 20 additions and 4 deletions.
24 changes: 20 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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

<details>
<summary>Args</summary>

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

</details>

```
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
```

0 comments on commit efc16f3

Please sign in to comment.