diff --git a/demos/yolo_nas/Dockerfile b/demos/yolo_nas/Dockerfile new file mode 100644 index 00000000..1add14a1 --- /dev/null +++ b/demos/yolo_nas/Dockerfile @@ -0,0 +1,20 @@ +# https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/tags +FROM nvcr.io/nvidia/pytorch:22.08-py3 + +RUN apt-get update && \ + apt-get install -y libgl1 && \ + apt-get clean && \ + rm -rf /var/lib/apt/lists/* + +RUN pip install --upgrade pip + +RUN pip install -r https://raw.githubusercontent.com/Deci-AI/super-gradients/master/requirements.txt && \ + # For some reason it doesn't work with latest version of OpenCV \ --> AttributeError: partially initialized module 'cv2' has no attribute '_registerMatType' (most likely due to a circular import) + pip install opencv-python==4.5.5.64 + +RUN pip install git+https://github.com/tryolabs/norfair.git@master#egg=norfair + +COPY requirements.txt requirements.txt +RUN pip install -r requirements.txt + +WORKDIR /demo/src/ \ No newline at end of file diff --git a/demos/yolo_nas/README.md b/demos/yolo_nas/README.md new file mode 100644 index 00000000..7aa6e68c --- /dev/null +++ b/demos/yolo_nas/README.md @@ -0,0 +1,22 @@ +# YOLO nas example + +Simplest possible example of tracking. Based on [YOLO-NAS-L](https://github.com/Deci-AI/super-gradients/blob/master/YOLONAS.md). + +## Instructions + +1. Build and run the Docker container with `./run_gpu.sh`. +2. Copy a video to the `src` folder. +3. Within the container, run with the default parameters: + + ```bash + python demo.py