nmvr commited on
Commit
0c593ea
1 Parent(s): 277be0a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -3
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
+ ```