yolo nas demo #274
yolo nas demo #274
Conversation
demos/yolo_nas/src/demo.py
Outdated
| MAX_DISTANCE: int = 10000 | ||
|
|
||
|
|
||
| import super_gradients |
There was a problem hiding this comment.
imports should be at the top, before the constants
| return detections | ||
|
|
||
|
|
||
| def yolo_detections_to_norfair_detections( |
There was a problem hiding this comment.
Not something to implement in this PR but to think about and maybe add to the backlog
We've used this function so often that I believe it should be included in Norfair's utils. Maybe rename it to xywh_detections_to_norfair and call it with yolo_detections.xywh
demos/yolo_nas/Dockerfile
Outdated
|
|
||
| RUN pip install git+https://github.com/tryolabs/norfair.git@master#egg=norfair | ||
|
|
||
| RUN pip install super-gradients |
There was a problem hiding this comment.
we should the requirements from requirements.txt or remove the file
| @@ -0,0 +1 @@ | |||
| super-gradient==3.1.1 | |||
|
Could we maybe add a mention of this demo in the main README.md? It could fit well in the adding tracking to different detectors section. |
I have just realized that we need to have It in the master's first in order to provide a link to the demo from the main README, so forget whatever I said there. |
Add a new demo using Yolo Nas as the object detector