50658b9a67
Add RegNet models and weights
2020-05-18 00:08:52 -07:00
1904ed8fec
Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups
2020-05-13 15:17:08 -07:00
17270c69b9
Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.
2020-05-12 21:59:34 -07:00
c4ca016656
Merge pull request #145 from rwightman/resnest
...
ResNeSt
2020-05-12 13:38:29 -07:00
208e7912f7
Missed one of the abalation model entrypoints, update README
2020-05-12 13:36:31 -07:00
9cc289f18c
Exclude EfficientNet-L2 models from test
2020-05-12 13:07:03 -07:00
2f884a0ce5
Add resnest14, resnest26, and two of the abalation grouped resnest50 models
2020-05-12 12:21:52 -07:00
f4cdc2ac31
Add ResNeSt models
2020-05-11 23:27:09 -07:00
f0eb021620
Replace all None by nn.Identity() in HRNet modules
2020-05-05 21:09:35 +03:00
8d8677e03b
Fix #139 . Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.
2020-05-04 12:14:43 -07:00
9c15d57505
added efficientnet pruned weights
2020-05-03 09:29:45 +03:00
ea300709f0
Fix model create fn not passing num_classes through. Fix #135
2020-05-01 18:37:03 -07:00
2c438c4e0d
Merge branch 'master' of github.com:rwightman/pytorch-models
2020-05-01 17:03:17 -07:00
2681a8d618
Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop
2020-05-01 17:00:21 -07:00
9590f301a9
Merge branch 'blur' of https://github.com/VRandme/pytorch-image-models into VRandme-blur
2020-05-01 15:19:57 -07:00
a5220ad3aa
Merge pull request #125 from Separius/patch-1
...
fix typo in eca
2020-05-01 15:11:56 -07:00
0834fbc01c
Move pruned model adapt strings to separate txt files. A few minor formatting alignment tweaks.
2020-05-01 15:09:20 -07:00
ba793f5c1c
Merge branch 'adding_ECA_resnet' of https://github.com/yoniaflalo/pytorch-image-models into yoniaflalo-adding_ECA_resnet
2020-04-29 16:43:02 -07:00
07f19dd699
added eca resnet
2020-04-29 19:09:55 +03:00
be7c784d21
Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference
2020-04-27 17:50:19 -07:00
e3a98171b2
Remove some CRLF that made it into TResNet merge
2020-04-27 13:23:41 -07:00
0004f37d25
Some TResNet cleanup.
...
* allow use of global pool arg, test-time-pooling
* clean checkpoints to just contain state dict, add 448 res checkpoints
* support DataParallel via lazy filter creation for JIT Downsample
* some minor formatting (mostly alignment) preferences
2020-04-27 13:21:05 -07:00
20290b5b56
fix typo in eca
2020-04-14 15:20:35 +04:30
ebf82b84ac
Merge pull request #122 from mrT23/master
...
TResNet models
2020-04-12 18:23:46 -07:00
bdb165a8a4
Merge changes in feature extraction interface to MobileNetV3
...
Experimental feature extraction interface seems to be changed
a little bit with the most up to date version apparently found
in EfficientNet class. Here these changes are added to
MobileNetV3 class to make it support it and work again, too.
2020-04-13 02:02:14 +03:00
8a63c1add8
finalizing
2020-04-12 19:08:35 +03:00
6209146738
TResNet models
2020-04-12 18:44:12 +03:00
1a8f5900ab
Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
2020-04-09 01:41:54 -07:00
06a50a94a8
Fix minor typos in create_attn.py and resnet.py
...
'eca'->'ceca'
and
doest not-> does not
2020-04-07 21:15:57 +09:00
1a9ab07307
One too many changes at a time, fix missing C
2020-04-07 21:12:08 +09:00
f17b42bc33
Blur filter no longer a buffer
2020-04-07 21:12:08 +09:00
6cdeca24a3
Some cleanup and fixes for initial BlurPool impl. Still some testing and tweaks to go...
2020-04-07 21:12:08 +09:00
acd1b6cccd
Implement Functional Blur on resnet.py
...
1. add ResNet argument blur=''
2. implement blur for maxpool and strided convs in downsampling blocks
2020-04-07 21:12:08 +09:00
ce3d82b58b
Update blurpool.py
...
clean up code for PR
2020-04-07 21:12:07 +09:00
3a287a6e76
Create blurpool.py
...
Initial implementation of blur layer.
currently tests as correct against Downsample of original github
2020-04-07 21:12:07 +09:00
c99a5abed4
Merge pull request #115 from rwightman/mobilenetv2-experiment
...
MobileNet-V2 experiments
2020-04-05 17:32:06 -07:00
e34074b4da
Add final weights for MobileNet-V2 experiments
2020-04-05 17:31:02 -07:00
a6a5565de7
Fiddling...
2020-03-19 10:20:20 -07:00
5a16c533ff
Add better resnext50_32x4d weights trained by andravin
2020-03-18 14:43:50 -07:00
bc998cad91
Experimenting with some MobileNetV2 variations to compare against EfficientNet-Lite
2020-03-18 13:54:06 -07:00
3406e582cf
Add EfficientNet-Lite results, update README
2020-03-18 13:12:30 -07:00
bd05258f7b
EfficientNet-Lite model added w/ converted checkpoints, validation in progress...
2020-03-17 23:31:45 -07:00
56e2ac3a6d
Merge pull request #94 from rwightman/lr_noise
...
Learning rate noise, MobileNetV3 weights, and activate MobileNetV3/EfficientNet weight init change
2020-02-29 20:41:05 -08:00
c16f25ced2
Add MobileNetV3 Large weights, results, update README and sotabench for merge
2020-02-29 20:37:20 -08:00
c60069c1eb
Annotate types on drop fns to avoid torchscript error
2020-02-27 09:30:23 -08:00
9fee316752
Enable fixed fanout calc in EfficientNet/MobileNetV3 weight init by default. Fix #84
2020-02-24 15:11:26 -08:00
43225d110c
Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args.
2020-02-18 14:00:26 -08:00
f1d5f8a6c4
Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
2020-02-18 13:58:30 -08:00
569419b38d
Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
2020-02-15 21:18:25 -08:00
53c47479c4
Batch validation batch size adjustment, tweak L2 crop pct
2020-02-15 20:37:04 -08:00
Ross Wightman
Ross Wightman
Vyacheslav Shults
AFLALO, Jonathan Isaac
Sepehr Sameni
Alexey Chernov
talrid
Chris Ha