Merge pull request #125 from bit-bots/feature/better_cfg_generator

Better cfg customization for custom datasets
bit-bots · Dec 3, 2024 · 6a3ce22 · 6a3ce22
2 parents d0393c1 + 4501066
commit 6a3ce22
Show file tree

Hide file tree

Showing 7 changed files with 159 additions and 797 deletions.
diff --git a/.gitignore b/.gitignore
@@ -15,3 +15,7 @@ checkpoints/
 logs/
 
 .python-version
+
+*.onnx
+*.pth
+**/tuning_records.json
diff --git a/README.md b/README.md
@@ -70,6 +70,14 @@ You can adjust the log directory using `--logdir <path>` when running `tensorboa
 #### Classes
 Add class names to `data/custom/yoeo_names.yaml`.
 
+Run the following command to adapt the model file (cfg) to the new number of classes:
+
+```bash
+poetry run yoeo-custiomize-cfg -c config/yoeo.cfg -d config/custom.data -o config/yoeo-custom.cfg
+```
+
+This changes the layers of the model to fit the number of classes in your dataset.
+
 #### Image Folder
 Move the images of your dataset to `data/custom/images/`.
-Original file line number
+Diff line change
@@ Expand Up / @@ -15,3 +15,7 @@ checkpoints/ @@
     logs/
     .python-version
+    *.onnx
+    *.pth
+    **/tuning_records.json