Update README.md
Browse files
README.md
CHANGED
|
@@ -37,15 +37,15 @@ The dataset was created under the research project of Grant Agency of Czech Repu
|
|
| 37 |
|
| 38 |
### Training Procedure
|
| 39 |
|
| 40 |
-
The training was done using Pytorch library, where CrossEntropyLoss() together with AdamW optimizer function.
|
| 41 |
|
| 42 |
### Results & Metrics
|
| 43 |
|
| 44 |
-
The overall accuracy of training for all classes reaches 99%, the mean intersection of union reaches 70%.
|
| 45 |
|
| 46 |
#### Hardware
|
| 47 |
|
| 48 |
-
The training was done using NVIDIA Quadro P4000.
|
| 49 |
|
| 50 |
#### Software
|
| 51 |
|
|
|
|
| 37 |
|
| 38 |
### Training Procedure
|
| 39 |
|
| 40 |
+
The training was done using Pytorch library, where CrossEntropyLoss() together with AdamW optimizer function.
|
| 41 |
|
| 42 |
### Results & Metrics
|
| 43 |
|
| 44 |
+
The dataset has 1207 images in resolution of 416x416 pixels together with 1207 masks. The overall accuracy of training for all classes reaches 99%, the mean intersection of union reaches 70%.
|
| 45 |
|
| 46 |
#### Hardware
|
| 47 |
|
| 48 |
+
The training was done using NVIDIA Quadro P4000 with CUDA support.
|
| 49 |
|
| 50 |
#### Software
|
| 51 |
|