From c4fc251b7e83166189903eae95b7f83546d7d300 Mon Sep 17 00:00:00 2001 From: Meshal Alamr <68873733+MeshalAlamr@users.noreply.github.com> Date: Sun, 22 Jan 2023 02:41:13 +0300 Subject: [PATCH] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index 799553d..94b20b5 100644 --- a/README.md +++ b/README.md @@ -38,6 +38,7 @@ git clone https://github.com/MeshalAlamr/smartathon-theme-1.git 3. Install the project requirements: ``` +cd smartathon-theme-1 pip install -r requirements.txt ``` @@ -47,6 +48,8 @@ python inference.py --model_name pattrn --segments 8 ``` This will generate the output csv file containing the labels and bounding boxes for the visual pollutions in the ``results`` folder. +Note: In case you faced memory issues during inference, try to increase the ``segments`` parameter. + ## Model Weights #### The final model weights can be found [here](https://github.com/MeshalAlamr/smartathon-theme-1/blob/main/models/pattrn/weights/best.pt).