From 73d98fe39c0408642f323cdfd91e542887682c81 Mon Sep 17 00:00:00 2001 From: Mathis Date: Wed, 30 Jul 2025 17:06:24 +0200 Subject: [PATCH 1/2] preprocessed images as on device --- model-training/datasets.dvc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/model-training/datasets.dvc b/model-training/datasets.dvc index ed3ea050..826a6290 100644 --- a/model-training/datasets.dvc +++ b/model-training/datasets.dvc @@ -1,6 +1,6 @@ outs: -- md5: 4b3a23bc5a894228cbb81bc71150732d.dir - size: 310669866 - nfiles: 2084 +- md5: c2440fbd2b7321eac76f0bc4b38ca27a.dir + size: 346598881 + nfiles: 2764 hash: md5 path: datasets From 4783c7df9d390c733c90507a27f48a9f2b6de491 Mon Sep 17 00:00:00 2001 From: Mathis Date: Wed, 30 Jul 2025 17:11:34 +0200 Subject: [PATCH 2/2] updated config files for preprocessed dataset --- ...augmentation_custom_loss_preprocessed.yaml | 44 +++++++++++++++++++ model-training/datasets.dvc | 4 +- 2 files changed, 46 insertions(+), 2 deletions(-) create mode 100644 model-training/configs/yolo11n_bike_with_augmentation_custom_loss_preprocessed.yaml diff --git a/model-training/configs/yolo11n_bike_with_augmentation_custom_loss_preprocessed.yaml b/model-training/configs/yolo11n_bike_with_augmentation_custom_loss_preprocessed.yaml new file mode 100644 index 00000000..8795e17f --- /dev/null +++ b/model-training/configs/yolo11n_bike_with_augmentation_custom_loss_preprocessed.yaml @@ -0,0 +1,44 @@ +--- +project_name: bike_aug_custom_loss_preprocessed_dataset +output_dir: runs/ +model: models/yolo11n.pt + +# for a full list of training and augmentation arguments see: +# https://docs.ultralytics.com/modes/train/#train-settings +train_args: + mode: train + task: detect + # yamllint disable-line rule:line-length + data: ./datasets/combined_preprocessed/YOLO/combined_preprocessed_yolo_dataset.yaml + epochs: 300 + patience: 200 + save: true + val: true + plots: true + rect: false + pretrained: true + cache: false + verbose: true + device: cuda + batch: 16 + nbs: 16 + multi_scale: true + workers: 4 + amp: true + dropout: 0.1 + classes: + - 0 + # Loss function parameters + dfl: 2.0 + kobj: 3.0 + box: 5.0 + cls: 0.3 + # Augmentation Parameters + hsv_s: 0.7 # hue + hsv_h: 0.015 # saturation + hsv_v: 0.5 # brightness + degrees: 180 + scale: 0.1 + flipud: 0.1 + fliplr: 0.5 + mosaic: 0.0 diff --git a/model-training/datasets.dvc b/model-training/datasets.dvc index 826a6290..ac59023c 100644 --- a/model-training/datasets.dvc +++ b/model-training/datasets.dvc @@ -1,6 +1,6 @@ outs: -- md5: c2440fbd2b7321eac76f0bc4b38ca27a.dir - size: 346598881 +- md5: c16bc40755d3b2237e70d6fc6cf9b85d.dir + size: 346598894 nfiles: 2764 hash: md5 path: datasets