Update README.md
Browse files
README.md
CHANGED
@@ -47,7 +47,7 @@ Default settings were used for other training hyperparameters (find more informa
|
|
47 |
|
48 |
Model training was performed using the following code:
|
49 |
|
50 |
-
```
|
51 |
from ultralytics import YOLO
|
52 |
|
53 |
# Use pretrained Yolo segmentation model
|
@@ -64,7 +64,7 @@ model.train(data=yaml_path, name='model_name', epochs=100, imgsz=640, workers=4,
|
|
64 |
|
65 |
Evaluation results using the validation dataset are listed below:
|
66 |
|Class|Images|Class instances|Box precision|Box recall|Box mAP50|Box mAP50-95|Mask precision|Mask recall|Mask mAP50|Mask mAP50-95
|
67 |
-
|
68 |
All|102|563|0.909|0.918|0.94|0.648|0.889|0.892|0.896|0.567
|
69 |
Paragraph|102|197|0.957|0.99|0.985|0.966|0.957|0.99|0.985|0.952
|
70 |
Marginalia|102|48|0.887|0.917|0.922|0.664|0.888|0.917|0.927|0.669
|
@@ -78,11 +78,11 @@ More information on the performance metrics can be found [here](https://docs.ult
|
|
78 |
If the model file `tuomiokirja_regions_04122023.pt` is downloaded to a folder `\models\tuomiokirja_regions_04122023.pt`
|
79 |
and the input image path is `\data\image.jpg', inference can be perfomed using the following code:
|
80 |
|
81 |
-
```
|
82 |
from ultralytics import YOLO
|
83 |
|
84 |
# Initialize model
|
85 |
-
model = YOLO(
|
86 |
-
prediction_results = model.predict(source
|
87 |
```
|
88 |
More information for available inference arguments can be found [here](https://docs.ultralytics.com/modes/predict/#inference-arguments).
|
|
|
47 |
|
48 |
Model training was performed using the following code:
|
49 |
|
50 |
+
```python
|
51 |
from ultralytics import YOLO
|
52 |
|
53 |
# Use pretrained Yolo segmentation model
|
|
|
64 |
|
65 |
Evaluation results using the validation dataset are listed below:
|
66 |
|Class|Images|Class instances|Box precision|Box recall|Box mAP50|Box mAP50-95|Mask precision|Mask recall|Mask mAP50|Mask mAP50-95
|
67 |
+
|:----|:----|:----|:----|:----|:----|:----|:----|:----|:----|:----|
|
68 |
All|102|563|0.909|0.918|0.94|0.648|0.889|0.892|0.896|0.567
|
69 |
Paragraph|102|197|0.957|0.99|0.985|0.966|0.957|0.99|0.985|0.952
|
70 |
Marginalia|102|48|0.887|0.917|0.922|0.664|0.888|0.917|0.927|0.669
|
|
|
78 |
If the model file `tuomiokirja_regions_04122023.pt` is downloaded to a folder `\models\tuomiokirja_regions_04122023.pt`
|
79 |
and the input image path is `\data\image.jpg', inference can be perfomed using the following code:
|
80 |
|
81 |
+
```python
|
82 |
from ultralytics import YOLO
|
83 |
|
84 |
# Initialize model
|
85 |
+
model = YOLO('\models\tuomiokirja_regions_04122023.pt')
|
86 |
+
prediction_results = model.predict(source='\data\image.jpg', save=True)
|
87 |
```
|
88 |
More information for available inference arguments can be found [here](https://docs.ultralytics.com/modes/predict/#inference-arguments).
|