gau-nernst/vision-toolbox
Toolbox for vision tasks. Pre-trained vision backbones on ImageNet with PyTorch Lightning 🚀
Python
Issues
- 0
Add lightweight ViT
#7 opened by gau-nernst - 0
- 3
Comparing backbone to original YOLOv5 repo
#6 opened by AlonZolfi - 2
csd = ckpt['model'].float().state_dict()
#5 opened by gav1n-cheung - 0
checklist
#1 opened by gau-nernst - 3
Imagenet pretrained
#3 opened by virilo - 1
Timm models
#4 opened by mikeseven - 4
Darknet derivative version support
#2 opened by zhiqwang