Update README.md
Browse files
README.md
CHANGED
@@ -30,6 +30,28 @@ This results in two distinct caveats:
|
|
30 |
|
31 |
To train the model, we used a data cohort from 2020, consisting of 333 images of Blue whiting, manually labelled using RoboFlow, and split it into Training (80%) / Validation (10%) / Holdout-Test (10%).
|
32 |
|
33 |
-
Additionally, after training, we performed inference on the cohorts from 2021 and 2022, consisting of a total of 2018 images, and visually validated their quality as they had no labels.
|
34 |
-
|
35 |
-
All images are the same size, 1280x960
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
|
31 |
To train the model, we used a data cohort from 2020, consisting of 333 images of Blue whiting, manually labelled using RoboFlow, and split it into Training (80%) / Validation (10%) / Holdout-Test (10%).
|
32 |
|
33 |
+
Additionally, after training, we performed inference on the cohorts from 2021 and 2022, consisting of a total of `2018` images, and visually validated their quality as they had no labels.
|
34 |
+
|
35 |
+
All images are the same size, `1280x960`. To keep the aspect ratio we define the image size parameter as `[1920x1080]`
|
36 |
+
|
37 |
+
These were the parameters used to train the model:
|
38 |
+
|
39 |
+
```bash
|
40 |
+
yolo detect train \
|
41 |
+
data=$data \
|
42 |
+
model=yolov8n.pt \
|
43 |
+
epochs=1000 \
|
44 |
+
imgsz=[1920,1080] \
|
45 |
+
rect=True \
|
46 |
+
batch=64 \
|
47 |
+
save=True \
|
48 |
+
save_period=1 \
|
49 |
+
cos_lr=True \
|
50 |
+
optimizer="auto" \
|
51 |
+
warmup_epochs=5 \
|
52 |
+
plots=True \
|
53 |
+
seed=42 \
|
54 |
+
device=0 \
|
55 |
+
project=otolith_detection \
|
56 |
+
name=2020_normal
|
57 |
+
```
|