From 6764c8ad07cce0143a1dc7612c9b38fbb48046d4 Mon Sep 17 00:00:00 2001 From: a Date: Sun, 27 Apr 2025 12:18:10 +0800 Subject: [PATCH 1/2] Remove the decorator rank_zero_only from the function validate_log_directory. --- yolo/utils/logging_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/yolo/utils/logging_utils.py b/yolo/utils/logging_utils.py index f60410d4..b9558a19 100644 --- a/yolo/utils/logging_utils.py +++ b/yolo/utils/logging_utils.py @@ -317,7 +317,6 @@ def log_model_structure(model: Union[ModuleList, YOLOLayer, YOLO]): console.print(table) -@rank_zero_only def validate_log_directory(cfg: Config, exp_name: str) -> Path: base_path = Path(cfg.out_path, cfg.task.task) save_path = base_path / exp_name From f7221001c98a60f43985078f3c5e4e0e809510d8 Mon Sep 17 00:00:00 2001 From: a Date: Sun, 27 Apr 2025 12:40:51 +0800 Subject: [PATCH 2/2] add description of dataset format and label format --- README.md | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/README.md b/README.md index 21689331..3ec05d12 100644 --- a/README.md +++ b/README.md @@ -56,6 +56,33 @@ These are simple examples. For more customization details, please refer to [Note ## Training +### Dataset Format + +1. structure +``` +{dataset1} +├── images +│   ├── test +│   │   └── 1.jpg +│   ├── train +│   │   └── 2.jpg +│   └── val +│   └── 3.jpg +└── labels + ├── test + │   └── 1.txt + ├── train + │   └── 2.txt + └── val + └── 3.txt +``` + +2. label format +id is integer, coordinates are scaled to 0 ~ 1 +``` +id xmin ymin xmax ymax +``` + To train YOLO on your machine/dataset: 1. Modify the configuration file `yolo/config/dataset/**.yaml` to point to your dataset.