Merge branch 'dev' into update-docs

Nota-NetsPresso · Dec 4, 2024 · 2b23343 · 2b23343
2 parents 631ca43 + d735df3
commit 2b23343
Show file tree

Hide file tree

Showing 13 changed files with 725 additions and 16 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,21 +3,21 @@
 ## New Features:
 
 - Add instance number logging feature for detection task by `@hglee98` in [PR 577](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/577)
-- Add ELAN (Efficient Layer Aggregation Networks) module by `@hglee98` in [PR 585](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/585)
 - Add Precision and Recall metric for detection task by `@hglee98` in [PR 579](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/579)
-- Add SPPCSPLayer and SPPELAN module by `@hglee98` in [PR 586](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/586)
+- Add YOLOv9 by `@hglee98` in [PR 585](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/585), [PR 586](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/586), [PR 592](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/592), [PR 593](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/593), [PR 595](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/595)
 
 ## Bug Fixes:
 
-- Fix error of checkpoint saving by `@hglee98` in [PR 580](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/580/files)
+- Fix error of checkpoint saving by `@hglee98` in [PR 580](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/580)
+- Fix example scripts by `@hglee98` in [PR 596](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/596)
 
 ## Breaking Changes:
 
 No changes to highlight.
 
 ## Other Changes:
 
-- Refactor RT-DETR and generalize CSPRepLayer and RepVGG block by `@hglee98` in [PR 581](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/581)
+- Refactor RT-DETR and generalize CSPRepLayer and RepVGG block by `@hglee98` in [PR 581](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/581), [PR 594](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/594)
 - Generalize 2d pooling layers and define as custom layer by `@hglee98` in [PR 583](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/583)
 - Unify bbox transformation and IoU computing methods by `@hglee98` in [PR 587](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/587)
 

diff --git a/README.md b/README.md
@@ -76,10 +76,10 @@ Then, train your model with your own configuraiton:
 
 ```bash
 python train.py\
-  --data config/data/beans.yaml\
+  --data config/data/huggingface/beans.yaml\
   --augmentation config/augmentation/classification.yaml\
   --model config/model/resnet/resnet50-classification.yaml\
-  --training config/training/classification.yaml\
+  --training config/training.yaml\
   --logging config/logging.yaml\
   --environment config/environment.yaml
 ```
@@ -88,10 +88,10 @@ Or you can start NetsPresso Trainer by just executing console script which has s
 
 ```bash
 netspresso-train\
-  --data config/data/beans.yaml\
+  --data config/data/huggingface/beans.yaml\
   --augmentation config/augmentation/classification.yaml\
   --model config/model/resnet/resnet50-classification.yaml\
-  --training config/training/classification.yaml\
+  --training config/training.yaml\
   --logging config/logging.yaml\
   --environment config/environment.yaml
 ```

diff --git a/src/netspresso_trainer/losses/detection/__init__.py b/src/netspresso_trainer/losses/detection/__init__.py
@@ -17,4 +17,5 @@
 from .retinanet import RetinaNetLoss
 from .rtdetr import DETRLoss
 from .yolo import YOLOFastestLoss
+from .yolov9 import YOLOv9Loss
 from .yolox import YOLOXLoss