pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...

Results 196 pytorch-image-models issues
Sort by recently updated
recently updated
newest added

**Is your feature request related to a problem? Please describe.** The implementation of Relative Positional Encoding are all included 0 paddings, which turns out to be not necessary. **Describe the...

enhancement

With the recently Pytorch 1.6, adding [SWA](https://pytorch.org/blog/pytorch-1.6-now-includes-stochastic-weight-averaging/) should be easier. Also, it improves the accuracy. ¿Are you going to add it? I know avg_checkpoint.py does something similar. However, having SWA...

help wanted

Addresses #599 Continuing the conversation from the issue here. I've added `BlurPool2d` as the default `aa_layer`. Should I change this to `AntiAliasDownsampleLayer`? I saw your comment [here](https://github.com/rwightman/pytorch-image-models/blob/9cc7dda6e5fcbbc7ac5ba5d2d44050d2a8e3e38d/timm/models/layers/blur_pool.py#L6) about wanting to...

**Is your feature request related to a problem? Please describe.** While doing benchmark on timm-models with `benchmark.py`, I tried the following two ways: 1. `python benchmark.py --model-list _models.txt -b 128`...

enhancement

Hey, I've been using the bits_and_tpu branch to train on TPU VMs, however when I try to run the inference script it needs a GPU, could the inference code be...

enhancement
help wanted

Prototype update for post-quantization support with torch.quantization. Tested with torch 1.6. Supported models: - efficientnet family - mobilenetv3 - rexnet Quantizable models and modifications made to support quantizations are located...

You have developed great schedulers in timm. One thing that makes them a bit challenging to integrate with usual pytorch pipelines and lightning is the 'epoch' argument in the 'step'...

enhancement

How do I apply this nice repo to Multi-Label Classification task ? That is , a image has multi labels,I'd like to get the each label's confidence. sound like add...

enhancement

There are several applications from creators of SWIN Transformer in [Object detection](https://github.com/SwinTransformer/Swin-Transformer-Object-Detection) and [Semantic segmentation](https://github.com/SwinTransformer/Swin-Transformer-Semantic-Segmentation). But Implementation a bit different from the original SWIN for image classification (BasicLayer has additional...

enhancement