Update README.md
Browse files
README.md
CHANGED
|
@@ -16,12 +16,12 @@ pipeline_tag: image-segmentation
|
|
| 16 |
|
| 17 |
## Overview
|
| 18 |
|
| 19 |
-
Модель построена на **YOLO26-
|
| 20 |
|
| 21 |
- **Input image size:** `896`
|
| 22 |
- **Number of classes:** `6`
|
| 23 |
|
| 24 |
-
Модель обучалась в [Kaggle-ноутбуке](https://
|
| 25 |
|
| 26 |
## Quick Start
|
| 27 |
|
|
@@ -41,7 +41,7 @@ repo_id = "mitbersh/car-damage-segmentation"
|
|
| 41 |
|
| 42 |
weights_path = hf_hub_download(
|
| 43 |
repo_id=repo_id,
|
| 44 |
-
filename="
|
| 45 |
)
|
| 46 |
|
| 47 |
model = YOLO(weights_path)
|
|
@@ -60,7 +60,7 @@ print(results)
|
|
| 60 |
|
| 61 |
```bash
|
| 62 |
yolo segment predict \
|
| 63 |
-
model=
|
| 64 |
source=path/to/car_image.jpg \
|
| 65 |
imgsz=896 \
|
| 66 |
conf=0.25 \
|
|
|
|
| 16 |
|
| 17 |
## Overview
|
| 18 |
|
| 19 |
+
Модель построена на **YOLO26-m**. Основные параметры:
|
| 20 |
|
| 21 |
- **Input image size:** `896`
|
| 22 |
- **Number of classes:** `6`
|
| 23 |
|
| 24 |
+
Модель обучалась в [Kaggle-ноутбуке](https://www.kaggle.com/code/brshtskmit/train-car-damage-segmentation-yolov26-m-cardd).
|
| 25 |
|
| 26 |
## Quick Start
|
| 27 |
|
|
|
|
| 41 |
|
| 42 |
weights_path = hf_hub_download(
|
| 43 |
repo_id=repo_id,
|
| 44 |
+
filename="damage_segmentation.pt"
|
| 45 |
)
|
| 46 |
|
| 47 |
model = YOLO(weights_path)
|
|
|
|
| 60 |
|
| 61 |
```bash
|
| 62 |
yolo segment predict \
|
| 63 |
+
model=damage_segmentation.pt \
|
| 64 |
source=path/to/car_image.jpg \
|
| 65 |
imgsz=896 \
|
| 66 |
conf=0.25 \
|