vision-toolbox
vision-toolbox copied to clipboard
Toolbox for vision tasks. Pre-trained vision backbones on ImageNet with PyTorch Lightning 🚀
For learning purpose - [x] ViT: https://arxiv.org/abs/2010.11929 - [x] MLP-Mixer: https://arxiv.org/abs/2105.01601 - [x] DeiT: https://arxiv.org/abs/2012.12877 - [x] DeiT-III: https://arxiv.org/abs/2204.07118 - [x] CaiT: https://arxiv.org/abs/2103.17239 - [x] Swin: https://arxiv.org/abs/2103.14030 - [ ]...
Hey, I've been trying to load the pre-trained weights from this repo into the original YOLOv5 (changing state dict key names, etc.). However, something does not align in the weights...
## TODO Backbones - [ ] HRNet [[Paper v1]](https://arxiv.org/abs/1902.09212) [[Paper v2]](https://arxiv.org/abs/1908.07919) [[code]](https://github.com/HRNet) [[mmdet]](https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/backbones/hrnet.py) - [ ] VOLO? [[paper]](https://arxiv.org/abs/2106.13112) - [x] PatchConvNet (training results are not good. port weights over to...