Merge pull request #3 from aswin-roman/main

Problem: The mlflow remote tracking and artifacts storage was not enabled
DRAIVE · Nov 3, 2022 · 965db4c · 965db4c
2 parents a9cf004 + fd782dc
commit 965db4c
Show file tree

Hide file tree

Showing 470 changed files with 23,946 additions and 652 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -1,3 +0,0 @@
-[submodule "src/yolov5"]
-	path = src/yolov5
-	url = https://github.com/ultralytics/yolov5.git

diff --git a/dvc.lock b/dvc.lock
@@ -13,20 +13,25 @@ stages:
       md5: d7f522c4eacd4240d3c597526495602a.dir
       size: 5400566
       nfiles: 75
-    - path: src/submod_git
-      md5: 1cc108f2272374517f05856983f547c2
-      size: 54
+    - path: data/train/labels/
+      md5: bc69a46932ce8be3d357eed00e98facb.dir
+      size: 5732
+      nfiles: 75
     - path: src/train_yolo.py
-      md5: fb34975d5a8ee1afd070cf23a9a5b875
-      size: 1246
+      md5: 1c7d1218e0f2795417ee963182f566e6
+      size: 1245
+    - path: src/yolov5/
+      md5: 9b5c9f5ba9d37757a2b8454529044a45.dir
+      size: 2048113
+      nfiles: 170
     params:
       params.yaml:
         train.batch_size: 4
         train.epochs: 1
     outs:
     - path: models/
-      md5: 5189dbd524cf87c6804413db9b440c8a.dir
-      size: 14429506
+      md5: d410dcb7ecafa911dee0e26b86d08116.dir
+      size: 14429513
       nfiles: 2
   install_requirements:
     cmd: pip install -r src/requirements_yolo_updated.txt --extra-index-url https://download.pytorch.org/whl/cu116
@@ -41,16 +46,21 @@ stages:
       md5: 63ffe70257ff2d32976a967293263177.dir
       size: 1128509
       nfiles: 21
+    - path: data/valid/labels/
+      md5: caad1a4c185e3c5b577e0769f58d897d.dir
+      size: 1655
+      nfiles: 21
     - path: models/
-      md5: 5189dbd524cf87c6804413db9b440c8a.dir
-      size: 14429506
+      md5: d410dcb7ecafa911dee0e26b86d08116.dir
+      size: 14429513
       nfiles: 2
-    - path: src/submod_git
-      md5: 1cc108f2272374517f05856983f547c2
-      size: 54
     - path: src/val_yolo.py
-      md5: 0b7d74c5df43084653ed00ab11adee60
-      size: 2141
+      md5: dcb406fac6603568e03cc770bb6c36c1
+      size: 2514
+    - path: src/yolov5/
+      md5: 9b5c9f5ba9d37757a2b8454529044a45.dir
+      size: 2048113
+      nfiles: 170
     params:
       params.yaml:
         validate.conf_thres: 0.3
@@ -63,7 +73,15 @@ stages:
       md5: d7f522c4eacd4240d3c597526495602a.dir
       size: 5400566
       nfiles: 75
+    - path: data/train/labels/
+      md5: bc69a46932ce8be3d357eed00e98facb.dir
+      size: 5732
+      nfiles: 75
     - path: data/valid/images/
       md5: 63ffe70257ff2d32976a967293263177.dir
       size: 1128509
       nfiles: 21
+    - path: data/valid/labels/
+      md5: caad1a4c185e3c5b577e0769f58d897d.dir
+      size: 1655
+      nfiles: 21
diff --git a/dvc.yaml b/dvc.yaml
@@ -4,24 +4,28 @@ stages:
     deps:
     - src/requirements_yolo_updated.txt
 
-  prepare:
-    cmd: python src/prepare.py
-    always_changed: true
-    outs: 
-    - src/submod_git
+  # prepare:
+  #   cmd: python src/prepare.py
+  #   always_changed: true
+  #   outs: 
+  #   - src/submod_git
 
   download:
     cmd: aws --endpoint-url ${download.url} s3 cp s3://${download.bucket}/${download.commit}/${download.path} data --recursive 
     outs: 
     - data/train/images/
     - data/valid/images/
+    - data/train/labels/
+    - data/valid/labels/
 
   train:
     cmd: python src/train_yolo.py
     deps:
     - src/train_yolo.py
-    - src/submod_git
+    # - src/submod_git
     - data/train/images/
+    - data/train/labels/
+    - src/yolov5/
     params:
     - train.epochs
     - train.batch_size
@@ -31,10 +35,13 @@ stages:
   validate:
     cmd: python src/val_yolo.py
     deps:
-    - src/submod_git
+    # - src/submod_git
     - src/val_yolo.py
     - data/valid/images/
+    - data/valid/labels/
     - models/
+    - src/yolov5/
     params:
     - validate.conf_thres
     - validate.iou_thres
+
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/artifacts/models/F1.png b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/artifacts/models/F1.png
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/artifacts/models/PR.png b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/artifacts/models/PR.png
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/artifacts/models/train_results.png b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/artifacts/models/train_results.png
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/meta.yaml b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/meta.yaml
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/Precision b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/Precision
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/Recall b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/Recall
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/mAP_IoU_.5-.95 b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/mAP_IoU_.5-.95
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/mAP_IoU_0.50 b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/metrics/mAP_IoU_0.50
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Confidence threshold b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Confidence threshold
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Dataset b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Dataset
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/IoU threshold b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/IoU threshold
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Training Batch size b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Training Batch size
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Training epochs b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/params/Training epochs
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.runName b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.runName
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.source.git.commit b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.source.git.commit
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.source.name b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.source.name
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.source.type b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.source.type
diff --git a/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.user b/mlruns/0/3051cb1be43545b4b0a0688d77b0164b/tags/mlflow.user
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/meta.yaml b/mlruns/0/3392129e5b9d452f90474d227101e35a/meta.yaml
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/Precision b/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/Precision
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/Recall b/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/Recall
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/mAP_IoU_.5-.95 b/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/mAP_IoU_.5-.95
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/mAP_IoU_0.50 b/mlruns/0/3392129e5b9d452f90474d227101e35a/metrics/mAP_IoU_0.50
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Batch size b/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Batch size
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Confidence threshold b/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Confidence threshold
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Dataset b/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Dataset
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Epochs b/mlruns/0/3392129e5b9d452f90474d227101e35a/params/Epochs
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/params/IoU threshold b/mlruns/0/3392129e5b9d452f90474d227101e35a/params/IoU threshold
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.runName b/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.runName
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.source.git.commit b/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.source.git.commit
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.source.name b/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.source.name
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.source.type b/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.source.type
diff --git a/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.user b/mlruns/0/3392129e5b9d452f90474d227101e35a/tags/mlflow.user
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/meta.yaml b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/meta.yaml
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/Precision b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/Precision
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/Recall b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/Recall
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/mAP_IoU_.5-.95 b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/mAP_IoU_.5-.95
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/mAP_IoU_0.50 b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/metrics/mAP_IoU_0.50
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Batch size b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Batch size
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Confidence threshold b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Confidence threshold
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Dataset b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Dataset
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Epochs b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/Epochs
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/IoU threshold b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/params/IoU threshold
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.runName b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.runName
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.source.git.commit b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.source.git.commit
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.source.name b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.source.name
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.source.type b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.source.type
diff --git a/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.user b/mlruns/0/34d9aa59116b419dbc97b3da404de4f5/tags/mlflow.user
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/meta.yaml b/mlruns/0/51a3125b60764abca361f28cee39397b/meta.yaml
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/params/Batch size b/mlruns/0/51a3125b60764abca361f28cee39397b/params/Batch size
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/params/Confidence threshold b/mlruns/0/51a3125b60764abca361f28cee39397b/params/Confidence threshold
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/params/Dataset b/mlruns/0/51a3125b60764abca361f28cee39397b/params/Dataset
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/params/Epochs b/mlruns/0/51a3125b60764abca361f28cee39397b/params/Epochs
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/params/IoU threshold b/mlruns/0/51a3125b60764abca361f28cee39397b/params/IoU threshold
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.runName b/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.runName
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.source.git.commit b/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.source.git.commit
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.source.name b/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.source.name
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.source.type b/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.source.type
diff --git a/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.user b/mlruns/0/51a3125b60764abca361f28cee39397b/tags/mlflow.user
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/meta.yaml b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/meta.yaml
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/Precision b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/Precision
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/Recall b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/Recall
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/mAP_IoU_.5-.95 b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/mAP_IoU_.5-.95
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/mAP_IoU_0.50 b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/metrics/mAP_IoU_0.50
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Confidence threshold b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Confidence threshold
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Dataset b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Dataset
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/IoU threshold b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/IoU threshold
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Training Batch size b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Training Batch size
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Training epochs b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/params/Training epochs
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.runName b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.runName
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.source.git.commit b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.source.git.commit
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.source.name b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.source.name
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.source.type b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.source.type
diff --git a/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.user b/mlruns/0/57ac2e922b1940f6be4f56cb94e2104c/tags/mlflow.user
diff --git a/mlruns/0/60d84818644746cba0ab7fb2d4220684/artifacts/models/F1_curve_val.png b/mlruns/0/60d84818644746cba0ab7fb2d4220684/artifacts/models/F1_curve_val.png
diff --git a/mlruns/0/60d84818644746cba0ab7fb2d4220684/artifacts/models/PR_curve_val.png b/mlruns/0/60d84818644746cba0ab7fb2d4220684/artifacts/models/PR_curve_val.png