huggingface/pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
PythonApache-2.0
Issues
- 1
[FEATURE] Addition of UniRepLKNet Backbones
#2330 opened by imr555 - 7
[FEATURE] Load model config AND weights from local folder (support locally cloned HF hub models)
#2338 opened by rwightman - 4
- 11
Oscillation in the train loss on multiple GPUs
#2339 opened by sinahmr - 0
- 2
- 1
[FEATURE] Could you introduce abstract classes or interfaces for clarity and easier extension?
#2273 opened by cgebbe - 2
[FEATURE] MaxVit implementation without using custom operators to be able to compile for tensorrt
#2329 opened by Skier23 - 1
- 1
[BUG] create_model "vit_intern300m_patch14_448"
#2327 opened by JarvisFei - 1
[BUG] model that has LayerScale architecture will have its '.gamma' renamed to '.weight' in transformers library resulting in incompatible loading
#2324 opened by Rkyzzy - 7
[FEATURE] Add model support and pretrained downloadable links for preresnet-18
#2289 opened by a2jinhee - 1
- 0
Zero reproducibility of results presented here
#2309 opened by svobora - 3
[FEATURE] Add InternViT
#2300 opened by seefun - 1
- 5
[BUG] AttributeError: module 'timm.models._registry' has no attribute 'get_pretrained_cfgs_for_arch'. Did you mean: 'get_pretrained_cfg_value'
#2303 opened by Benjamin-Tan - 0
HOW CAN I Download vit_huge_patch14_224_in21k.pth
#2301 opened by twisti14 - 1
There are many model naming duplication issues
#2295 opened by jixiedy - 1
[BUG] Wrong model.num_features for mobilenetv4
#2290 opened by MichaelMonashev - 1
[BUG] Cannot load pretrained ConvNext model
#2287 opened by viet2411 - 2
- 0
[FEATURE] Kolmogorov–Arnold Transformer
#2283 opened by Adamdad - 2
- 2
- 0
[FEATURE] LowFormer Support
#2279 opened by ogencoglu - 0
[FEATURE] Model request: LW-DETR
#2278 opened by Immortalin - 1
[BUG] Incorrect string 'map ' with an extra ' '?
#2272 opened by sry2 - 1
- 0
- 2
[BUG] where are the OOD results?
#2259 opened by tfriedel - 1
ModuleNotFoundError: No module named 'timm.models.rdnet'
#2260 opened by SamMohel - 5
[FEATURE] solve `weights_only=False` warning,
#2249 opened by GaoShanwen - 0
- 3
[FEATURE] Support RDNet
#2241 opened by ogencoglu - 6
Gradient computation fails for 'HRNet_FeatureExtractor' due to an inplace operation
#2248 opened by mohammadalihumayun - 0
[FEATURE] Change default normalization ?
#2250 opened by asusdisciple - 5
Hiera can't use smaller img_size
#2247 opened by claverru - 1
convnext_tiny model produces outputs that are dependent on the batch size
#2244 opened by Avishka-Perera - 11
[BUG] out_indices are not always checked
#2242 opened by f-fuchs - 5
- 1
[BUG] CoAtNet_0 Model different from paper
#2235 opened by karam-nus - 1
- 0
- 3
[BUG] MobileNetV4 Output Shape Is Wrong
#2228 opened by samils7 - 2
inference error
#2218 opened by the-cat-crying - 1
[BUG] Transform not properly working with batching and Grayscale Images in ViT
#2219 opened by asusdisciple - 1
Accuracy keeps the same for several epoches
#2222 opened by qiuzhewei - 1
The latest MobileNetV4 hyrid models are not inlcuded.
#2221 opened by yuanhaoguo - 2