Skip to content

Commit

Permalink
update README
Browse files Browse the repository at this point in the history
  • Loading branch information
OneMagicKey committed Jun 16, 2024
1 parent 80efc0a commit 5789d9c
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -127,13 +127,13 @@ follow these steps:
folder, depending on the model type
3) Add the model to the `model_list` in `src/bot.py` using the following format:
`ModelInfo(model_type, model_name, model_input_size, model_version)`
`ModelInfo(type, name, input_size, conf, iou, version)`
```python
# Example:
ModelInfo("detection", "yolov8s", (640, 640), 8), # small detection model
ModelInfo("segmentation", "yolov8m-seg", (640, 480), 8), # medium segmentation model with rectangular input size
ModelInfo("detection", "yolov8s", (640, 640), version=8), # small detection model
ModelInfo("segmentation", "yolov8m-seg", (640, 480), version=8), # medium segmentation model with rectangular input size
```
## References
Expand Down

0 comments on commit 5789d9c

Please sign in to comment.