PyTorch官网机器视觉骨干网络合集生成器 WebUI

本仓库汇聚了 PyTorch 官方网站上所有机器视觉预训练模型的骨干网络集合。主要包括在 ImageNet1K 数据集上进行预训练的多种卷积神经网络（CNN）和 Vision Transformer 等视觉模型。整个集合被划分为 V1 和 V2 两个子集，涵盖了多个经典和先进的视觉模型版本。这些预训练骨干网络为用户提供了在图像识别、目标检测、图像分割等任务中进行迁移学习的强大基础。同时，它也为研究者和从业者在不同场景中灵活应用这些预训练模型提供了便捷的选择。