mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
2,448 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

2448 Commits (a50713ce6e55c90dde085f049c35ec8a695447c4)
 

Author SHA1 Message Date
Ross Wightman a50713ce6e Fix #2272 2024-09-02 13:20:05 -07:00
Ross Wightman ebbe530ee4 Add MobileNetV3 RA4 (mnv4 recipe) weights 2024-09-02 13:10:34 -07:00
Ross Wightman b3f3a005a0 Add RTX4090 inference benchmark numbers for latest models 2024-08-26 17:15:05 -07:00
Ross Wightman 072783331f Version 1.0.9 2024-08-23 11:47:42 -07:00
Ross Wightman 76b0e9931a Placeholder for new mnv3 model 2024-08-23 10:11:20 -07:00
Ross Wightman ed7aaf8d6d
Merge pull request #2264 from huggingface/group_size_eff 
Allow group_size override for more efficientnet and mobilenetv3 based…
 2024-08-22 15:24:02 -07:00
Ross Wightman bef0c124cc Add re-run of imagenet val/ood test set results. Fixed wrong crop for models w/ diff test sizes. Add latest sbb2 and ra4 weights. 2024-08-22 11:56:05 -07:00
Ross Wightman 39e92f0c0d mobilenet_edgetpu can use group_size override, more consistency in arg wrap/sadface w/ extra group_size arg 2024-08-22 11:44:02 -07:00
Ross Wightman b9f020a509 Allow group_size override for more efficientnet and mobilenetv3 based models 2024-08-21 16:51:38 -07:00
Ross Wightman 26ed96b052
Create UPGRADING.md 2024-08-21 15:58:54 -07:00
Ross Wightman 29d69e8734
Update README.md 2024-08-21 15:53:04 -07:00
Ross Wightman 9f228b7fc4
Update README.md 2024-08-21 15:41:54 -07:00
Ross Wightman 00c5be7656
Merge pull request #2258 from huggingface/sbb2_vit_hiera_weights 
Update Hiera model for abswin & add more in12k weights for hiera & vit
 2024-08-21 12:50:15 -07:00
Ross Wightman 17923a66bb Add layer scale to hieradet 2024-08-21 11:23:39 -07:00
Ross Wightman 47e6958263 Add hierdet_small (non sam) model def 2024-08-21 11:05:54 -07:00
Ross Wightman 076efef73b
Merge pull request #2263 from andravin/fix-docs-install-dir 
Fix install directory in hfdocs.
 2024-08-21 10:24:16 -07:00
Ross Wightman 9fcbf39cdc Add remaining sbb vit betwixt/mediumd fine-tunes 2024-08-21 10:09:38 -07:00
Ross Wightman dc94cca0e5 Remaining Hiera sbb weights uploaded 2024-08-21 10:06:27 -07:00
Andrew Lavin 3970558e0d Fix install directory in hfdocs. 
Change from "timm" to "pytorch-image-models"
 2024-08-21 09:27:52 -07:00
Ross Wightman a256e50457 Move padding back in front of windowing 2024-08-17 11:22:53 -07:00
Ross Wightman 7d83749207 pool size test fixes 2024-08-17 08:27:13 -07:00
Ross Wightman 1bd92bca0e Add fused_attn flag to HieraDet attn block 2024-08-16 22:57:49 -07:00
Ross Wightman 691bb54443 Larger min input size needed 2024-08-16 17:09:19 -07:00
Ross Wightman de3a91a7a0 Add min_input_size of 128 for hieradet/sam2 2024-08-16 15:13:56 -07:00
Ross Wightman 0b05122cda Fixing hieradet (sam2) tests 2024-08-16 14:33:40 -07:00
Ross Wightman e035381171 Move padding out of windowing code for hieradet, fix torchscript typing issues, make pooling MaxPool unique instances across two modules 2024-08-16 13:36:33 -07:00
Ross Wightman 146c2fbe34 Add resnet50d and efficientnet_b1 ra4 (mnv4) hparam weights 2024-08-16 12:10:00 -07:00
Ross Wightman 962958723c More Hiera updates. Add forward_intermediates to hieradat/sam2 impl. Make both use same classifier module. Add coarse bool to intermediates. 2024-08-16 11:10:04 -07:00
Ross Wightman f2cfb4c677 Add WIP HieraDet impl (SAM2 backbone support) 2024-08-15 17:58:15 -07:00
Ross Wightman a50e53d41f Rename global pos embed for Hiera abswin, factor out commonly used vit weight init fns to layers. Add a channels-last ver of normmlp head. 2024-08-15 17:46:36 -07:00
Ross Wightman 2f3fed43b8 Fix hiera init with num_classes=0, fix weight tag names for sbb2 hiera/vit weights, add LayerScale/LayerScale2d to layers 2024-08-15 11:14:38 -07:00
Ross Wightman fee91fdd41 Update Hiera model for abswin, more stable weight init, layer-scale. ImageNet-12k weights for hiera_small_abswin, and two of the sbb vits with improved reg4 init. 2024-08-14 12:22:40 -07:00
Ross Wightman ac3470188b Add mobilenetv1_125 weights from ra4 / mnv4 hparam series 2024-08-14 08:47:02 -07:00
Ross Wightman 907555fda6 Update model img-size/crop expansion for bulk runner 2024-08-13 10:42:09 -07:00
Ross Wightman 4515a435e4 Update imagenet & OOD test set eval results with current weights. Now have both test & train resolution results in tables. 2024-08-13 10:25:25 -07:00
Ross Wightman 2df9f2869d
Merge pull request #2257 from huggingface/weights_only 
Add weights only flag to avoid warning, try to keep bwd compat.
 2024-08-12 18:09:36 -07:00
Ross Wightman a7b0bfc376 exception for unknown arg should have been TypeError 2024-08-12 16:09:36 -07:00
Ross Wightman bd0f79153b Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249 2024-08-12 11:41:41 -07:00
Ross Wightman 531215eded Bring changelog up to current, back to dev version, 1.0.9.dev0 2024-08-08 09:17:23 -07:00
Ross Wightman d29b720d8d Add changes to hfdocs toctree 2024-08-08 09:12:58 -07:00
Ross Wightman 4fb878bdd6 Remove old docs 2024-08-08 09:08:56 -07:00
Ross Wightman f96e369615
Update changes.md 2024-08-08 08:37:49 -07:00
Ross Wightman 33ab2b35ae
Update README.md 2024-08-08 08:36:04 -07:00
Ross Wightman bf781bc3c2
Merge pull request #2245 from dhkim0225/feature/rdnet 
Feature: Add RDNet
 2024-08-07 16:19:22 -07:00
Ross Wightman 49f6e1da96 Remove eval to avoid safety concerns 2024-08-07 14:54:13 -07:00
Ross Wightman 2d5c9bf60d
Merge pull request #2253 from brendanartley/hgnet-grad-checkpointing 
Add gradient checkpointing to hgnets
 2024-08-07 12:45:14 -07:00
Ross Wightman e9ef9424f0 Add a few missing __all__ entries. 2024-08-07 09:35:51 -07:00
brendanartley 232a459e36 Added gradient checkpointing to hgnet 2024-08-07 09:22:51 -07:00
dong-hyun ea33d6a74d bugfix pool_size 2024-08-03 09:50:20 +09:00
Ross Wightman 10344625be proj_drop wasn't being called in sam vit 2024-08-02 16:12:02 -07:00