#8438: Yolov4 demo

tenstorrent · May 15, 2024 · 70469ec · 70469ec
1 parent 9531535
commit 70469ec
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,16 @@
+# 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.
+- Weights can be downloaded from [here](https://drive.google.com/file/d/1wv_LiFeCRYwtpkqREPeI13-gPELBDwuJ/view) to load wieghts from local directory.
+
+## To run the demo:
+1. Checkout to branch `ankit/ttnn_yolov4_demo`.
+2. Run `pytest models/experimental/functional_yolov4/demo/ttnn_demo_yolov4.py`.
+3. Prediction image will be generated at TT_METAL_HOME (current) directory.