pytorch-image-models:基于 PyTorch 的图像模型库项目

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

分支209Tags78

项目介绍

Hugging Face的PyTorch图像模型库(huggingface/pytorch-image-models)是由Hugging Face团队开发并持续维护的一个开源项目,该库集成了众多性能卓越的预训练模型,旨在为图像识别、分类等计算机视觉任务提供强大的工具支持。【此简介由AI生成】

定制我的领域
32037.07 K5.19 K访问 GitHub