- Add yolov9 pretrained weights by
@illian01in PR 631 - Add EXIR exporting feature by
@illian01in PR 632
No changes to highlight.
No changes to highlight.
Fix/add data params mlflow by @hglee98 in PR 629
- Add num_save_samples option and code refactoring by
@illian01in PR 625
- Fix classification postprocessing bug by
@illian01in PR 627
No changes to highlight.
- Refactoring sample saving logic by
@illian01in PR 626 - Fix dockerfile torch version by
hglee98in PR 624
- Upload ONNX model to mlflow server if mlflow logging option is enabled by
hglee98in PR 621
- Fix bug where training failed with torch.fx detection model by
hglee98in PR 619
No changes to highlight.
- Loosen torch version coverage, not fixed version by
@illian01in PR 622
- Add MLFlow train tracking feature by
@hglee98in PR 615
No changes to highlight.
No changes to highlight.
No changes to highlight.
- Add cache_data option by
@hglee98in PR 608
- Fix augmentation config of yolox training example by
@hglee98in PR 610
- Upgrade torch to 2.0.1 by
@illian01in PR 613
- Delete label_size, targets parameters from forward method across models by
@hglee98in PR 611
- Add instance number logging feature for detection task by
@hglee98in PR 577 - Add Precision and Recall metric for detection task by
@hglee98in PR 579 - Add YOLOv9 by
@hglee98in PR 585, PR 586, PR 592, PR 593, PR 595, PR 589, PR 590, PR 597, PR 598, PR 601, PR 602 - Add prediction values to evaluation summary by
@hglee98in PR 600
No changes to highlight.
- Refactor RT-DETR and generalize CSPRepLayer and RepVGG block by
@hglee98in PR 581, PR 594 - Generalize 2d pooling layers and define as custom layer by
@hglee98in PR 583 - Unify bbox transformation and IoU computing methods by
@hglee98in PR 587 - Update documents by
@hglee98in PR 591 - Update mosaic augmentation epoch specification by
@hglee98in PR 604
- Add RandomResize2 by
@illian01in PR 550 - Add confidence score on detection visualization by
@hglee98in PR 552 - Add save_best_only option for saving model by
@hglee98in PR 555, PR 567 - Add option to select best model saving criterion by
@hglee98and@illian01in PR 557, PR 573, PR 574 - Add MultiStepLR scheduler by
@hglee98in PR 559 - Add class-wise metric analysis option by
@illian01in PR 568 - Add ReLU6 by
@hglee98in PR 566 - Add TFLite model evaluation feature by
@hglee98in PR 563 - Add YOLO-Fastest-v2 by
@hglee98in PR 548 - Add tabulating step for metric standard outputs by
@illian01in PR 570
- Fix keyword error of segmentation training by
@illian01in PR 551 - Fix typo when saving optimizer state_dict by
@hglee98in PR 553 - Fix not initialized save_dtype error by
@hglee98in PR 565 - Fix mAP error in case of certain classes object is not in the dataset
@hglee98in PR 571, PR 572
- Massive refactoring metric modules and add flexible metric selecting option by
@illian01in PR 564
No changes to highlight.
- Add no_weight_decay and overwrite option in optimizer config by
@illian01in PR 534 - Fuse reparameterable layers before save by
@hglee98in PR 531 - Add ONNX model evaluation feature by
@illian01in PR 539 - Add ONNX export tool by
@illian01in PR 541 - Add VOC 2012 dataset auto-downloader by
@illian01in PR 546
- Fix bug in RandomCrop and segmentation dataset getitem by
@illian01in PR 535 - Handle unexpected error during training rt-detr by
@hglee98in PR 530 - Fix no 'deploy' attribute error in fx model train phase by
@illian01in PR 543 - Fix bugs of PIDNet forward and saving by
@illian01in PR 544
- Remove fx_model_path in model configuration and parsing model format by file extensions by
@illian01in PR 538
- Document updates by
@illian01in PR 533 - Delete install command for the optional requirements from Dockerfile by
@hglee98in PR 532
- Add RT-DETR by
@illian01and@hglee98in PR 490, PR 491, PR 494, PR 498, PR 500, PR 507, PR 501 - Add Objects365 dataset auto downloader by
@hglee98in PR 482 - Add ResNet model parameters to support various form by
@illian01in PR 497, PR 523 - Add RandomIoUCrop, RandomZoomOut augmentation by
@hglee98in PR 504 - Add MobileNetV4 backbone by
@illian01andhglee98in PR 516, PR 520, PR 526 - Add gradient clipping feature by
hglee98in PR 506 - Enabled to control
ToTensorandNormalizationthrough config file by@illian01in PR 522 - Enable to set onnx opset version through config file by
@illian01in PR 525
- Fix handling error in case of error occured in first epoch by
@illian01in PR 493 - Fix error in FLOPs computation by
@illian01in PR 499
No changes to highlight.
- Update pi 4b deployment benchmark by
@illian01in PR 492 - Clamp bbox for detection dataset loading by
@hglee98in PR 503 - Combine requirements and requirements-optional by
@illian01in PR 517
- Add YOLOX-nano and YOLOX-tiny by
@hglee98in PR 467 - Separate postprocessor configuration hierarchy by
@hglee98in PR 470 - Add YOLO-Fastest by
@hglee98in PR 471 - Write detailed status on training_summary by
@illian01in PR 487
No changes to highlight.
No changes to highlight.
- Change attention bias interpolate method by
@illian01in PR 468 - Fix ViT token number in positional encoding for torch.fx compile step by
@illian01in PR 475 - Remove output typing of PIDNet by
@illian01in PR 477 - Update documentation by
@illian01in PR 483, PR 485 - Minor refactorings by
hglee98in PR 513, PR 518
- Update Benchmarks & Checkpoints (docs) and weights files to fully usable by
@illian01in PR 446, PR 447, PR 456, PR 461 - Add TFLite runtime code example by
@illian01in PR 449
- Fix best_epoch init error in TrainingSummary in case of training resume by
@illian01in PR 448 - Fix segmentation metric logic bug by
@illian01in PR 455, PR 460
No changes to highlight.
- Refactoring: remove thop, replace MACs with FLOPs by
@illian01in PR 444 - Add copyright for entire project by
@illian01in PR 451 - Modify ImageSaver to receive various resolution at once
@illian01in PR 458
- Add dataset validation step and refactoring data modules by
@illian01in PR 417, PR 419 - Add various dataset examples including automatic open dataset format converter by
@illian01in PR 430 - Allow using text file path for the
id_mappingfield by@illian01in PR 432, PR 435
- Fix test directory check line by
@illian01in PR 428 - Fix Dockerfile installation commandline
@cbpark-notain PR 434
No changes to highlight.
- Save training summary at every end of epochs by
@illian01in PR 420 - Refacotring: rename postprocessors/register.py to registry.py by
@aychunin PR 424 - Add example configuration set by
@illian01in PR 438 - Documentation: fix simple use config file path by
@cbpark-notain PR 437
- Add activation and dropout layer in FC by
@illian01in PR 325, PR 327 - Add function to Resize: Match longer side with input size and keep ratio by
@illian01in PR 329 - Add transforms: MosaicDetection by
@illian01in PR 331, PR 337, PR 397 - Add transform: HSVJitter by
@illian01in PR 336, PR 413 - Add transforms: RandomResize by
@illian01in PR 341, PR 344, PR 398 - Add model EMA (Exponential Moving Average) by
@illian01in PR 348 - Add entry point for evaluation and inference by
@illian01in PR 374, PR 379, PR 381, PR 383 - Add classification visulizer by
@illian01in PR 384 - Add dataset caching feature by
@illian01in PR 391 - Add mixed precision training by
@illian01in PR 392 - Add YOLOX l1 loss activation option by
@illian01in PR 396 - Add NetsPresso Trainer YOLOX pretrained weights by
@illian01in PR 406
- Fix output_root_dir from fixed string to config value by
@illian01in PR 323 - Gather predicted results before compute metric and fix additional distributed evaluation inaccurate error by
@illian01in PR 346, PR 356 - Fix detection score return by
@illian01in PR 373 - Fix memory leak from onnx export by
@illian01in PR 386, PR 394 - Refactoring metric modules and fix inaccurate metric bug by
@illian01in PR 402
- Simplify augmentation configuration hierarchy by
@illian01in PR 322 - Add pose estimation task and RTMPose model by
@illian01in PR 357, PR 366 - Remove pythonic config and move training initialization functions to
trainer_main.pyby@illian01in PR 371 - Unify gradio demo in one page by
@deepkyuin PR 408
- Refactoring: Move custom transforms to each python module by
@illian01in PR 332 - Update Pad transform to receive target size of image by
@illian01in PR 334 - Rafactoring: Fix to make transform object in init by
@illian01in PR 339 Add before_epoch step which does update modules like dataloader before epoch training by@illian01in PR 340- Revert PR 340 and add multiprocessing.Value to handle MosaicDetection and RandomResize by
@illian01in PR 345 - Enable adjust max epoch of scheduler by
illian01in PR 350 - Remove github action about hugging face space demo by
@illian01in PR 351 - Update docs by
@illian01in PR 355, PR 410 - Backbone task compatibility checking refactoring by
@illian01in PR 361, PR 364 - Fix postprocessor return type as numpy.ndarray by
@illian01in PR 365 - Update default asignees of issue template by
@illian01in PR 375 - Refactoring: Remove CSV logger, change logger module input format by
@illian01in PR 377 - Change ClassficationDataSampler logic by
@illian01in PR 382 - Add YOLOX weights initialization step by
@illian01in PR 393 - Minor update: detection postprocessor, dataset, and padding strategy by
@illian01in PR 395 - Specify input size for onnx export and remove augmentation.img_size by
@illian01in PR 399 - Update issue and pr template by
@illian01in PR 401 - Add documentation auto deploy action by
@illian01in PR 405
No changes to highlight.
- Remove union of int and list by
@illian01in PR 317
No changes to highlight.
No changes to highlight.
- Enable customizing inference transform by
@illian01in PR 304 - Add transform function: CenterCrop by
@illian01in PR 308
- Fix automatic PIDNet weights download bug by
@illian01in PR 306 - Resize default value to list by
@illian01in PR 315
No changes to highlight.
- Update model caching directory and checkpoint configuration by
@deepkyuin PR 299, PR 312 - Minor docs update by
@illian01in PR 300 - Update software development stage by
@illian01in PR 301 - Fix size param of Resize to receive int or list by
@illian01in PR 310 - Modify PIDNet conv bias, add head_list property on models by
@illian01in PR 311
- Construct head by config file by
@illian01in PR 237 - Construct neck by config file by
@illian01in PR 249 - Add model: RetinaNet by
@illian01in PR 257 - Select
gpuswithenvironmentconfiguration by@deepkyuin PR 269 - Return logging directory path and fix training interfaces by
@deepkyuin PR 271 - Add transform: AutoAugment by
@illian01in PR 281
- Fix attribute error on fc by
@illian01in PR 252 - Restore file export for stream log by
@deepkyuin PR 255 - Fix CSV logging, configuration error, and misused loggings by
@deepkyuin PR 259 - Fix minor bug in train.py by
@illian01in PR 277 - Fix local classification dataset loader error by
@illian01in PR 279 - Fix safetensors file overwriting bug by
@illian01in PR 289 - Fix error on full model load by
@illian01in PR 295
- Provide pytorch state dict with
.safetensorsand training summary with.jsonfor a better utilization by@deepkyuin PR 262
- Refactoring for detection models by
@illian01in PR 260 - Equalize logging format with
PyNetsPressoby@deepkyuin PR 263 - Refactoring for clean docs by
@illian01in PR 265, PR 266, PR 272, PR 273, PR 274, PR 284 - Update docs up-to-date by
@illian01in PR 278 - Refactoring model building code and move TASK_MODEL_DICT by
@illian01in PR 282 - Add eps param on RMSprop by
@illian01in PR 285 - Fix weights loading logic by
@illian01in PR 287, PR 290 - Change pretrained checkpoint name convention and update weight path and url by
@illian01in PR 291 - Move seed field to environment config by
@illian01in PR 292 - Move ResNet and Fc implementation code to core directory by
@illian01in PR 294
- Add a gpu option in
train_with_config(only single-GPU supported) by@deepkyuin PR 219 - Support augmentation for classification task: cutmix, mixup by
@illian01in PR 221 - Add model: MixNet by
@illian01in PR 229 - Add
model.nameto get the exact nickname of the model by@deepkyuin PR 243 - Add transforms: RandomErasing and TrivialAugmentationWide by
@illian01in PR 246
- Fix PIDNet model dataclass task field by
@illian01in PR 220 - Fix default criterion value of classification
@illian01in PR 238 - Fix model access of 2-stage detection pipeline to compat with distributed environment by
@illianin PR 239
- Enable dataset augmentation customizing by
@illian01in PR 201 - Add postprocessor module by
@illian01in PR 223 - Equalize the model backbone configuration format by
@illian01in PR 228 - Separate FPN and PAFPN as neck module by
@illian01in PR 234 - Auto-download pretrained checkpoint from AWS S3 by
@deepkyuin PR 244
- Update ruff rule (
W) by@deepkyuin PR 218 - Integrate classification loss modules by
@illian01in PR 226
- Add YOLOX model by
@illian01in PR 195, PR 212 - Fix Faster R-CNN detection head to compat with PyNP compressor by
@illian01in PR 184, PR 194, PR 204 - Support multi-GPU training with
netspresso-trainentrypoint by@deepkyu,@illian01and@Only-bottlein PR 213
- Remove fx training flag in entry point by
@illian01in PR 188 - Fix bounding box coordinates computing error on random flip augmentation by
@illian01in PR 211
- Release NetsPresso Trainer colab tutorial
@illian01in PR 191 - Support training with python-level config by
@deepkyuin PR 205
- Refactoring models/op module by
@illian01in PR 189, PR 190 - Parameterize activation function of BasicBlock and Bottleneck by
@illian01in PR193 - Modify MobileNetV3 to stage format and remove forward hook by
@illian01in PR 199 - Substitute MACs counter with
fvcorelibrary to sync with NetsPresso by@deepkyuand@Only-bottlein PR 202 - Enable to compute metric with all training samples by
@illian01in PR 210
- Add MobileNetV3 by
@illian01in PR 173 - Handling for fp16 fx model by
@illian01in PR 175 - Deploy Gradio simulators to Hugging Face Space by
@deepkyuin PR 181
No changes to highlight.
No changes to highlight.
- Removed
model_namecheck increate_transform_segmentationfunction by@illian01in Pr 176 - Combine entry point to
train.pyby@illian01in Pr 180
No changes to highlight.
⚠️ Fix pypi package import by@deepkyuin PR 169
No changes to highlight.
No changes to highlight.
- Support RGB segmentation map and class with label value by
@deepkyuin PR 163
- Fix import error for
Sequenceby@illian01in PR 155 - Add last epoch validation and delete save_converted_model by
@illian01in PR 157
No changes to highlight.
- Add onnx save in best model saving step of graph model training by
@illian01in PR 160. - Update to keep community standards by
@illian01in PR 162 - Update a lot of contents in docs (but not finished...) by
@deepkyuin PR 165 - Add github workflow for pypi packaging by
@deepkyuin PR 166
Notice: there are some changes in maintaining the repository and we transferred the original private repository to the public(planned) location. Some PR links may be expired because those links are based on the previous version of repository. Hope you understand.
This change is applied at PR 151
- Update the model configuration to handle the architecture by
@deepkyuin PR 130
- Add
tensorboardin requirements.txt by@illian01in PR 134 - Fix typo in
scripts/example_train.shby@illian01in PR 137 - Initialize loss and metric at same time with optimizer and lr schedulers by
@deepkyuin PR 138 - Hotfix the error which shows 0 for validation loss and metrics by fixing the variable name by
@deepkyuin PR 140 - Add missing field,
save_optimizer_state, inlogging.yamlby@illian01in PR 149 - Hotfix for pythonic config name (classification loss) by
@deepkyuin PR 242
- Add checkpoint saving while training, resume training with the checkpoint, save the training summary with # Params and MACs by
@deepkyuin PR 135 - Change parsing argument for FX model retraining and resuming training to model configuration by
@deepkyuin PR 135 - Apply ruff linter and add workflow for ruff checking by
@deepkyuin PR 143
- Add PyNetsPresso tab in documentation page by
@deepkyuin PR 128 - Fix issue template and default assignee per issue type by
@deepkyuin PR 144
- Generalize segmentation head and add support ResNet50 + segmentation by
@deepkyuin PR 122
No changes to highlight.
No changes to highlight.
- Simplify training configuration and example training scripts by
@deepkyuin PR 124 - Add
PyNetsPressotab in docs page by@deepkyuin PR 128
- Add LR simulator (powered by gradio) with
trainingconfiguration by@deepkyuin[PR 116](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/116) - Add augmentation simulator (powered by gradio) with
augmentationconfiguration by@deepkyuin[PR 118](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/118) - Add LR scheduler (cosine with warm restart, step_lr) by
@deepkyuin[PR 114](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/114)
No changes to highlight.
- Support detection training with its metric by
@deepkyuin[PR 119](https://github.com/Nota-NetsPresso/netspresso-trainer/pull/119)
No changes to highlight.