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).