Advertisement

PyTorch图像分类模型及相关脚本、预训练权重集合(python源码)。

  •  5星
  •     浏览量: 0
  •     大小:None
  •      文件类型:None


简介:
PyTorch 的图像模型、脚本以及预训练权重涵盖了广泛的选择,包括 (SE)ResNet/ResNeXT、DPN、EfficientNet、MixNet、MobileNet-V3/V2/V1、MNASNet、Single-Path NAS、FBNet 等等。这些资源于 2020 年 6 月 11 日进行了更新。 此次更新包含一系列重要的改进和新增功能:DenseNet 模型得到了优化,修复了一个先前存在的内存效率问题;同时,增加了模糊池和深干添加 (VoVNet) 的支持,并训练了 39 个 V2 变体(ese_vovnet_39b),其 top-1 激活率达到了 79.3%。此外,还引入了新的激活函数,在模型构建阶段提供选择权,从而能够更灵活地适应脚本或跟踪工具(例如 ONNX 导出)。 为了增强模型的可用性,实验性地添加了内存高效的梯度计算 (grad) 以及 ME hard_swish 函数。 上下文管理器 (context mgr) 被引入,用于控制可导出的状态,支持导出到脚本或禁用 JIT 编译。 DenseNet 和 VoVNet 模型现在也具备初始的上下文层支持。

全部评论 (0)

还没有任何评论哟~
客服
客服
  • PyTorchPython
    优质
    该文集汇集了使用PyTorch进行图像分类的各种资源,包括多个深度学习模型架构、实用脚本及多种预训练权重文件的Python实现代码。 PyTorch 图像模型的最新更新(2020年6月11日)包括一系列改进:DenseNet 模型采用了来自 Torchvision 的内存高效添加功能以修复一个错误,新增了模糊池化以及深干路径的优化;VoVNet V1 和 V2 模型也进行了训练,并且在79.3 top-1精度上表现出了ese_vovnet_39b模型的良好性能。此外,还引入了一些新的激活函数功能:用户可以在创建模型时选择不同的操作方式以提高灵活性和与脚本或跟踪兼容的激活(ONNX 导出),其中包括实验性的 hard_mish 激活,并且添加了内存高效的 grad 和 ME hard_swish 上下文管理器。为了进一步优化,还加入了 Norm + Activation 组合层在 DenseNet 和 VoVNet 中的支持以进行初步试验。
  • PyTorch - (SE)ResNet/ResNeXt,DPN,EfficientNet,MixNe...
    优质
    本文介绍了使用PyTorch实现的一系列先进的图像处理模型,包括(SE)ResNet/ResNeXt,DPN,EfficientNet和MixNet等,并提供了详细的脚本和预训练权重。 PyTorch图像模型包括(SE)ResNet/ResNeXT、DPN、EfficientNet、MixNet、MobileNet-V3/V2/V1、MNASNet以及单路径NAS和FBNet等,最近更新了DenseNet模型,并提高了内存效率以修复错误。此外还添加了模糊池和深茎功能,增加了VoVNet V1和V2模型的版本,并将ese_vovnet_39b变体训练至79.3 top-1激活工厂。新的激活操作可以在创建模型时执行,以便在使用脚本或跟踪(ONNX导出)以及hard_mish(实验性)兼容的激活时更加灵活。
  • VOC格式的火灾检测Yolo_Tiny
    优质
    本项目提供了一套基于VOC格式的火灾检测图像数据集,并附有针对YOLO-Tiny模型优化后的预训练权重文件,旨在促进火灾监控系统的研发与应用。 通过处理将火焰图集合转换为VOC数据集,并使用训练好的yolo_tiny火焰检测weights进行训练后,可以实现基于yolo-tiny的图像检测功能。
  • Yolov8目标检测
    优质
    本资源集合提供了一系列基于YOLOv8框架的目标检测预训练模型权重文件,适用于各类图像识别任务。 YOLOv8的五个权重文件可供下载。由于从GitHub下载速度较慢,我已将这些文件上传至个人空间以方便大家获取。
  • PyTorch SRCNN与测试代
    优质
    本资源提供基于PyTorch框架实现的SRCNN模型训练与测试代码以及预训练权重文件,适用于图像超分辨率任务研究。 基于PyTorch平台的用于图像超分辨率的深度学习模型SRCNN包括网络模型、训练代码、测试代码、评估代码以及预训练权重。评估代码可以计算在RGB和YCbCr空间下的峰值信噪比(PSNR)和结构相似度。
  • Inception_V1_PyTorch: PyTorch中的Inception_V1实现
    优质
    简介:Inception_V1_PyTorch是基于PyTorch框架实现的GoogLeNet(Inception_v1)模型,提供预训练权重下载和使用。适合图像分类任务。 `inception_v1.pytorch` 是一个在 PyTorch 上使用预训练权重实现 Inception V1 的代码。这段代码是 Soumith 火炬仓库中的 PyTorch 版本翻译:它实现了原始架构的初始版本,即著名的 GoogLeNet。可以在 ImageNet 数据集上找到该模型的预训练权重,并且测试精度为 26.38%。 如果我没记错的话,这是在原始火炬回购中遇到的一个问题——数据加载正确性的问题。如果您能够通过训练此模型获得更高的准确性,请提供新的权重给我!该代码根据 MIT 许可证发布。
  • 使用 PyTorch ResNet50 进行Python实现
    优质
    本项目通过Python语言及PyTorch框架,利用预训练ResNet50模型高效地实现了图像分类功能。提供详尽代码示例,助力快速理解和应用深度学习技术于视觉任务中。 代码说明: 数据预处理:定义了训练集和验证集的数据预处理操作,包括随机裁剪、水平翻转、归一化等。 数据集加载:使用 torchvision.datasets.ImageFolder 加载数据集,并通过 DataLoader 创建数据加载器。 模型加载与修改:加载预训练的 ResNet50 模型,冻结预训练层的参数,修改最后一层全连接层以适应自定义的分类类别数。 训练模型:定义了训练函数 train_model,在训练过程中使用交叉熵损失函数和随机梯度下降优化器。 保存模型:在完成训练后,将模型权重保存到 image_classification_model.pth 文件中。 预测部分:加载已保存的模型权重,并通过 predict_image 函数对单张图片进行预测,最后显示预测结果。 使用说明: 确保你的数据集按照 hymenoptera_data 目录结构组织,包含 train 和 val 子目录,每个子目录下的文件夹代表一个类别。将 test_image.jpg 替换为你实际要预测的图片路径。
  • ResNet_50_New:基于ResNet 50
    优质
    ResNet_50_New是一款利用了ResNet 50模型预训练权重进行高效图像分类的深度学习代码,适用于快速部署与定制化开发。 resnet_50_new:使用ResNet 50权重进行图像分类。
  • Yolov10程序
    优质
    简介:本资源提供YOLOv10深度学习目标检测算法的预训练模型权重文件及其配套程序代码,便于用户快速部署和二次开发。 Yolov10的预训练权重以及其训练测试程序包括了用于模型训练和测试的所有必要代码及官方预训练权重文件。提供的预训练权重有多种版本:yolov10b.pt、yolov10l.pt、yolov10m.pt、yolov10n.pt、yolov10s.pt 和 yolov10x.pt,此外还有包含所有必要资源的压缩文件 yolov10-main.zip。
  • YOLOv5
    优质
    简介:YOLOv5预训练模型的权重是基于深度学习的目标检测算法,提供高效准确的图像中目标识别与定位功能,适用于多种应用场景。 YOLOv5 提供了四个不同大小的预训练模型权重文件:yolov5s.pt、yolov5m.pt、yolov5l.pt 和 yolov5x.pt。这些文件分别代表小(S)、中(M)、大(L)和特大(X)版本,适用于不同的计算资源和精度需求。