pytorch-image-models
pytorch-image-models copied to clipboard
huggingface
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...
”FasterViT: Fast Vision Transformers with Hierarchical Attention“ https://github.com/NVlabs/FasterViT  The code is written based on timm and provides pretrained weights on ImageNet1k. But there are many layers customized in the...
Thanks for your wonderful work! Given now that PyTorch-XLA 2.1 is just released, and many new features are added compared to PyTorch-XLA 1.xx, I wonder if you have any plan...
tkx for your wonderful work, I found the `timm` is not support shufflenet v1 and v2 yet, there is any plan to support it? and [here](https://github.com/megvii-model/ShuffleNet-Series) is the official repository...
Using https://github.com/NVIDIA/TransformerEngine to speed up transformer based models on new Nvidia Hopper GPUs and float8 training. Ideally it will detect that you are using Ada based GPUs and adapt the...
HRFormer implementation would be great :) https://github.com/HRNet/HRFormer
**Issue Description:** Sometimes, we need to apply consistent transformations to multiple images, particularly in tasks like semantic segmentation. In such cases, both the input image and its corresponding GT map...
This pull request add [SimpleNet architecture](https://arxiv.org/abs/1608.06037). Simplenetv1 is a 2016 architecture comprised of only the most basic operators that comprises a plain CNN network. It outperformed many deeper and more...
Hello! I'm Caleb, and I am engineer working on research/open source at Comet ML. Recently, we've had several requests for an integration between Comet and timm. Would it be alright...
**Describe the bug** VOLO models can not be exported to ONNX: ``` ==> Creating PyTorch volo_d1_224 model .../lib/python3.8/site-packages/safetensors/torch.py:99: UserWarning: TypedStorage is deprecated. It will be removed in the future and...
As above, the paper seems a lot better than seaformer, which is already very good.
Metadata
Owner
Metadata
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...