#8438: Yolov4 demo

tenstorrent · May 16, 2024 · 6e67431 · 6e67431
1 parent e36a8f5
commit 6e67431
Show file tree

Hide file tree

Showing 29 changed files with 3,788 additions and 3 deletions.
diff --git a/models/experimental/functional_yolov4/demo/README.md b/models/experimental/functional_yolov4/demo/README.md
@@ -0,0 +1,15 @@
+# Yolov4 ttnn Demo
+
+## Inputs required:
+
+
+- A demo input image 320x320.
+- coco.names which has class names.
+- current branch  - `ankit/ttnn_yolov4_demo`.
+- Input image and class names file available under `tests/ttnn/integration_tests/yolov4/`
+- Pretrained weights are loaded from weka path.
+- If weka path is not available weights can be downloaded from [here](https://drive.google.com/file/d/1wv_LiFeCRYwtpkqREPeI13-gPELBDwuJ/view)  and place it to `tests/ttnn/integration_tests/yolov4/yolov4.pth`.
+## To run the demo:
+1. Checkout to branch `ankit/ttnn_yolov4_demo`.
+2. Run `pytest models/experimental/functional_yolov4/demo/demo.py`.
+3. Prediction image will be generated at TT_METAL_HOME (current) directory.