[WIP] crop dataset by 300px

classification_model_training/files_from_makeathon/data/test.dvc

@@ -1,6 +1,6 @@
 outs:
-- md5: e03ff535d9f3da7f51ed642b70ec5d95.dir
-  size: 549163
+- md5: c6f373f6997ea2bae313d494caa35b43.dir
+  size: 1434598
   nfiles: 80
   hash: md5
   path: test

classification_model_training/files_from_makeathon/data/train.dvc

@@ -1,6 +1,6 @@
 outs:
-- md5: 0263429cb9fefbc8b68b6a88c26d942f.dir
-  size: 2506709
+- md5: c6c8f93cb47471d0133562e34c3f56d0.dir
+  size: 6053320
   nfiles: 362
   hash: md5
   path: train

classification_model_training/files_from_makeathon/dvc.lock

@@ -97,3 +97,47 @@ stages:
       hash: md5
       md5: 49a445cb95476dbc8dc7e7e2530050f0
       size: 74
+  evaluate-last:
+    cmd: python predict.py --model efficientnet-b2 --image_size 224 --num_classes
+      4 --dataset ./data/test/images --gt ./data/test/gt.csv --single_model_path models/model.pt
+    deps:
+    - path: data/test
+      hash: md5
+      md5: e03ff535d9f3da7f51ed642b70ec5d95.dir
+      size: 549163
+      nfiles: 80
+    - path: models/model.pt
+      hash: md5
+      md5: b7e4e7f26989ba76d1009d1a78724b9c
+      size: 31270857
+    - path: predict.py
+      hash: md5
+      md5: 43fb163a6a3546d1bd6b47cf92fc6ff5
+      size: 6001
+    params:
+      params.yaml:
+        base:
+          random_seed: 1
+        evaluate:
+          dataset: ./data/test/images
+          gt: ./data/test/gt.csv
+          device: cuda
+          val_bs: 16
+        train:
+          fold: 1
+          dataset: ./data/train/images
+          gt: ./data/train/gt.csv
+          num_classes: 4
+          metric: f1_score
+          device: cuda
+          epochs: 30
+          train_batch: 16
+          val_batch: 16
+          outdir: models
+          lr: 0.001
+          dropout_rate: 0.3
+          drop_connect_rate: 0.2
+          batch_norm_momentum: 0.99
+          batch_norm_epsilon: 0.001
+          model: efficientnet-b2
+          image_size: 224

classification_model_training/files_from_makeathon/dvc.yaml

@@ -59,3 +59,20 @@ stages:
     metrics:
     - models/metrics-evaluate.json:
         cache: false
+  evaluate-last:
+    cmd: >-
+      python predict.py
+      --model ${train.model}
+      --image_size ${train.image_size}
+      --num_classes ${train.num_classes}
+      --dataset ${evaluate.dataset}
+      --gt ${evaluate.gt}
+      --single_model_path ${train.outdir}/model.pt
+    deps:
+    - data/test
+    - predict.py
+    - models/model.pt
+    params:
+    - base
+    - train
+    - evaluate