Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
MeshalAlamr authored Jan 21, 2023
1 parent 907fe37 commit c4fc251
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
```

Expand All @@ -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.

<b>Note</b>: In case you faced memory issues during inference, try to increase the ``segments`` parameter.


## Model Weights <a name="model" />
#### The final model weights can be found [here](https://github.com/MeshalAlamr/smartathon-theme-1/blob/main/models/pattrn/weights/best.pt).
Expand Down

0 comments on commit c4fc251

Please sign in to comment.