Universal feature extraction, new models, new weights, new test sets.
Universal feature extraction, new models, new weights, new test sets.
* All models support the `features_only=True` argument for `create_model` call to return a network that extracts features from the deepest layer at each stride.
* All models support the `features_only=True` argument for `create_model` call to return a network that extracts features from the deepest layer at each stride.
* New models
* New models
* CSPResNet, CSPResNeXt, CSPDarkNet, DarkNet
* CSPResNet, CSPResNeXt, CSPDarkNet, DarkNet
@ -14,6 +15,8 @@ Universal feature extraction, new models, new weights, new test sets.
* DPN68b - 79.2 top-1
* DPN68b - 79.2 top-1
* EfficientNet-Lite0 (non-TF ver) - 75.5 (submitted by @hal-314)
* EfficientNet-Lite0 (non-TF ver) - 75.5 (submitted by @hal-314)
* Add 'real' labels for ImageNet and ImageNet-Renditions test set, see [`results/README.md`](results/README.md)
* Add 'real' labels for ImageNet and ImageNet-Renditions test set, see [`results/README.md`](results/README.md)
* Train script and loader/transform tweaks to punch through more aug arguments
* README and documentation overhaul. See initial (WIP) documentation at https://rwightman.github.io/pytorch-image-models/
The model architectures included come from a wide variety of sources. Sources, including papers, original impl ("reference code") that I rewrote / adapted, and PyTorch impl that I leveraged directly ("code") are listed below.
The model architectures included come from a wide variety of sources. Sources, including papers, original impl ("reference code") that I rewrote / adapted, and PyTorch impl that I leveraged directly ("code") are listed below.
Most included models have pretrained weights. The weights are either:
Most included models have pretrained weights. The weights are either:
1. from their original sources
1. from their original sources
2. ported by myself from their original impl in a different framework (e.g. Tensorflow models)
2. ported by myself from their original impl in a different framework (e.g. Tensorflow models)
3. trained from scratch using the included training script
3. trained from scratch using the included training script
@ -55,16 +56,17 @@ The validation results for the pretrained weights can be found [here](results.md