Refactor openmixup.core (instead of openmixup.hooks) and openmixup.models.augments (contains mixup augmentation methods which are originally implemented in openmixup.models.utils). After code refactoring, the macro design of OpenMixup is similar to most projects of MMLab.
Support deployment of ONNX and TorchScript in openmixup.core.export and tools/deployment. We refactored the abstract class BaseModel (implemented in openmixup/models/classifiers/base_model.py) to support forward_inference (for custom inference and visualization). We also refactored openmixup.models.heads and openmixup.models.losses to support forward_inference. You can deploy the classification models in OpenMixup according to deployment tutorials.
Support testing API methods in openmixup/apis/test.py for evaluation and deployment of classification models.
Refactor openmixup.core.optimizers to separate optimizers and builders and support the latest Adan optimizer.
Supporting new features:
Support detailed usage instructions in README of config files for image classification methods in configs/classification, e.g., mixups on ImageNet. We will update READMEs of other methods in configs/selfsup and configs/semisup in a few weeks.
Refine the origianzation of README files according to README-Template.
Support the new mixup augmentation method (AlignMix) and provide the relevant config files in various datasets.
Updating documents:
Update documents of mixup benchmarks on ImageNet in Model_Zoo_sup.md. Update config files for supported mixup methods.
Update formats (figures, introductions and content tables) of awesome lists in Awesome Mixups and Awesome MIM and provide the latest methods.
Code refactoring:
openmixup.core
(instead ofopenmixup.hooks
) andopenmixup.models.augments
(contains mixup augmentation methods which are originally implemented inopenmixup.models.utils
). After code refactoring, the macro design ofOpenMixup
is similar to most projects of MMLab.ONNX
andTorchScript
inopenmixup.core.export
andtools/deployment
. We refactored the abstract classBaseModel
(implemented inopenmixup/models/classifiers/base_model.py
) to supportforward_inference
(for custom inference and visualization). We also refactoredopenmixup.models.heads
andopenmixup.models.losses
to supportforward_inference
. You can deploy the classification models inOpenMixup
according to deployment tutorials.openmixup/apis/test.py
for evaluation and deployment of classification models.openmixup.core.optimizers
to separate optimizers and builders and support the latest Adan optimizer.Supporting new features:
configs/classification
, e.g., mixups on ImageNet. We will update READMEs of other methods inconfigs/selfsup
andconfigs/semisup
in a few weeks.Updating documents: