首页 > 社交 > 科普中国

用户实践|使用

常驻编辑 科普中国 2022-09-30 青云   模型   服务器   样本   架构   显卡   深度   硬盘   数据   用户   技术

本文作者:张宁 64J拜客生活常识网

浙江华双信息科技有限公司 技术总监64J拜客生活常识网

负责带领技术团队进行重点产品、项目研发,负责系统架构设计,规划平台未来技术架构方向。目前主要研究领域为图像识别、目标检测、语义分割,曾负责“三防三白”中工作服、帽子及口罩佩戴检测预警、洪水预报预警等业务。64J拜客生活常识网

操作场景64J拜客生活常识网

使用 TeslaA100 GPU 云服务器进行 YOLOv5 模型训练用于目标检测任务,查看训练效果。64J拜客生活常识网

YOLOv5 模型简介64J拜客生活常识网

YOLOv5 模型是 Ultralytics 公司于 2020 年 6 月 9 日公开发布的。YOLOv5 模型是基于 YOLOv3 模型基础上改进而来的,由 Backbone、Neck 和 Head 组成。64J拜客生活常识网

示意图如下:64J拜客生活常识网

示例环境64J拜客生活常识网

  • 实例类型:单卡TeslaA100 GPU 云服务器64J拜客生活常识网

  • 系统盘:50 GB高性能云硬盘64J拜客生活常识网

  • 操作系统:Ubuntu20.0464J拜客生活常识网

  • 带宽:20 M(按流量计费)64J拜客生活常识网

  • 本地操作系统:MacOS64J拜客生活常识网

64J拜客生活常识网

PytochGPU 环境搭建

参考青云官方文档《UbuntuGPU 云服务器搭建深度学习环境》中的“GPU 云服务器显卡相关配置”和“Pytorch 安装”两大模块。64J拜客生活常识网

64J拜客生活常识网

训练数据

训练数据集选用 coco128,一共 128 张图片,一张图对应一个标注好的 Label。64J拜客生活常识网

数据集概览如下:64J拜客生活常识网

64J拜客生活常识网

训练数据上传服务器

通过 Linux 连接工具,选择 sftp,输入对应的服务器公网 IP、账号、密码即可连接云服务器。64J拜客生活常识网

64J拜客生活常识网

训练结果

利用 GPU 进行训练,每个 Epoch 在 1 秒内完成,300 轮 Epoch 总耗时 7 分钟,Percision 和 Recall 达到 90% 多,IOU0.5 情况下,mAP 接近 1。64J拜客生活常识网

每个Epoch在1秒内完成,300轮Epoch总耗时7分钟64J拜客生活常识网

Percision 和 Recall 达到 90% 多,IOU0.5 情况下,mAP 接近 164J拜客生活常识网

查看训练结果:64J拜客生活常识网

64J拜客生活常识网

补充测试

coco128 训练数据集较小,使用另一份较大的训练数据集做测试,图片约 8000 张,大小 790 M,每轮 epoch 在 40s 内完成,表现很好。64J拜客生活常识网

每轮 epoch 在 40s 内完成64J拜客生活常识网

张 宁64J拜客生活常识网

浙江华双信息科技有限公司 64J拜客生活常识网

技术总监64J拜客生活常识网

“青云 GPU 云服务器使用下来很惊喜,TeslaA100 显卡很给力,非常适合深度学习相关模型的训练。64J拜客生活常识网

而且,按需计费模式下,GPU 云服务器关机后,云服务器的 CPU、内存、GPU 卡这些计算资源不会收费,只有系统盘仍会产生费用,把系统盘的容量缩小就可以节省成本了。64J拜客生活常识网

如果训练样本过大,想以比较低的成本传输训练样本数据,可以创建一个硬盘,在 GPU 云服务器关机的时候,将硬盘卸载,然后创建一个低配置的云服务器,将硬盘再挂载这个低配服务器上进行样本数据传输,等训练时将硬盘再挂载至 GPU 云服务器进行训练。”64J拜客生活常识网

64J拜客生活常识网

青云最新 GPU 云服务器申请试用

搭载 Ampere 架构 NVIDIAA100 Tensor Core GPU 显卡,单卡显存 40 GB,单台云服务器最多可挂载 8 块 GPU,适用于深度学习、科学计算应用加速。64J拜客生活常识网

即刻申请,新用户可获得 200优惠券!64J拜客生活常识网

贡献实践文档还可获得 500-1000大额优惠券奖励!64J拜客生活常识网

相关阅读:

  • 穷且益坚不坠青云之志的意思意思,穷且益坚不坠青云之志
  • 五四青年节金句
  • 北京大兴出现1例确诊病例 青云店镇枣林村实施封控
  • 大兴出现一名确诊病例,青云店镇枣林村实施封控
  • 解封不解防!大兴区青云店镇枣林村正式解除封控管控
  • 凭借东风力,扶摇上青云
  • 你好,六月
  • 高考从来不是改变人生的一次决定,而是足够年轻时的一次
  • 北京夏天玩水好去处,乐谷银滩真的可
  • 闲夏人慵懒,来自张三的哲学时光
    • 网站地图 |
    • 声明:登载此文出于传递更多信息之目的,并不意味着赞同其观点或证实其描述。文章内容仅供参考,不做权威认证,如若验证其真实性,请咨询相关权威专业人士。