pytorch-image-models/models
Ross Wightman 4663fc2132 Add support for tflite mnasnet pretrained weights and included spnasnet pretrained weights of my own.
* tensorflow 'SAME' padding support added to GenMobileNet models for tflite pretrained weights
* folded batch norm support (made batch norm optional and enable conv bias) for tflite pretrained weights
* add url for spnasnet1_00 weights that I recently trained
* fix SE reduction size for semnasnet models
2019-04-28 17:45:07 -07:00
..
__init__.py Uniform pretrained model handling. 2019-04-11 21:32:16 -07:00
adaptive_avgmax_pool.py Fix pooling in mnasnet, more sensible default for AMP opt level 2019-04-17 18:06:37 -07:00
conv2d_same.py Add support for tflite mnasnet pretrained weights and included spnasnet pretrained weights of my own. 2019-04-28 17:45:07 -07:00
densenet.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
dpn.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
genmobilenet.py Add support for tflite mnasnet pretrained weights and included spnasnet pretrained weights of my own. 2019-04-28 17:45:07 -07:00
helpers.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
inception_resnet_v2.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
inception_v4.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
median_pool.py Initial commit, puting some ol pieces together 2019-02-01 22:07:34 -08:00
model_factory.py Add support for tflite mnasnet pretrained weights and included spnasnet pretrained weights of my own. 2019-04-28 17:45:07 -07:00
pnasnet.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
resnet.py Add checkpoint clean script, add link to pretrained resnext50 weights 2019-04-13 14:15:35 -07:00
senet.py Add seresnet26_32x4d cfg and weights + interpolation str->PIL enum fn 2019-04-14 13:43:46 -07:00
test_time_pool.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00
xception.py Add per model crop pct, interpolation defaults, tie it all together 2019-04-12 22:55:24 -07:00