mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
75 Commits 141 Branches 65 Tags
Commit Graph

13 Commits

Author SHA1 Message Date
Ross Wightman
a4516fe0fb Add pretrained weights for FBNet-C 2019-05-20 11:05:25 -07:00
Ross Wightman
76539d905e Some transform/data/loader refactoring, hopefully didn't break things 
* factor out data related constants to own file
* move data related config helpers to own file
* add a variant of RandomResizeCrop that randomizes interpolation method
* remove old Numpy version of RandomErasing
* cleanup torch version of RandomErasing and use it in either GPU loader batch mode or single image cpu Transform
 2019-05-16 22:52:17 -07:00
Ross Wightman
db8ad25a23 MobileNetV3 appears correct based on paper update, cleaned up comments and compacted last block def 2019-05-15 08:53:27 -07:00
Ross Wightman
c3fbdd4655 Fix efficient head for MobileNetV3 2019-05-11 11:23:11 -07:00
Ross Wightman
17da1adaca A few MobileNetV3 tweaks 
* fix expansion ratio on early block
* change comment re stride mistake in paper
* fix rounding not being called properly for all multipliers != 1.0
 2019-05-11 10:23:40 -07:00
Ross Wightman
6523e4abe4 More appropriate name for se channel 2019-05-10 23:32:18 -07:00
Ross Wightman
db056d97e2 Add MobileNetV3 and associated changes hard-swish, hard-sigmoid, efficient head, etc 2019-05-10 23:28:13 -07:00
Ross Wightman
02abeb95bf Add the url for tflite mnasnet ported weights 2019-04-28 23:21:35 -07:00
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
Ross Wightman
afb357ff68 Make genmobilenet weight init switchable, fix fan_out in google style linear init 2019-04-22 17:46:17 -07:00
Ross Wightman
34cd76899f Add Single-Path NAS pixel1 model 2019-04-22 12:43:45 -07:00
Ross Wightman
419555be62 Update a few GenMobileNet comments 2019-04-22 12:30:55 -07:00
Ross Wightman
bc264269c9 Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related 
* add an alternate RMSprop opt that applies eps like TF
* add bn params for passing through alternates and changing defaults to TF style
 2019-04-21 15:54:28 -07:00