当前位置: 首页 > news >正文

基于PyTorch的交通标志目标检测系统

一、开发环境

  • Windows 10
  • PyCharm 2021.3.2
  • Python 3.7
  • PyTorch 1.7.0

二、制作交通标志数据集,如下图

三、配置好数据集的地址,然后开始训练

python train.py --data traffic_data.yaml --cfg traffic_yolov5s.yaml --weights pretrained/yolov5s.pt --epoch 100 --batch-size 4

四、训练完成后进行推理预测,效果如图

五、完整源码、数据集和模型文件下载

链接:https://pan.baidu.com/s/111wLXWLckTfrHIEGZyXeqA?pwd=w5q6
提取码:w5q6

http://www.lryc.cn/news/154210.html

相关文章:

  • feign调用失败 feign.RetryableException: xxx-service executing GET http://xxx/test
  • mysql 用户管理
  • pyinstaller打包exe运行闪退
  • ARM 汇编基础知识
  • CRM 自动化如何改善销售和客户服务?
  • Bean 的六种作用域
  • go语言--锁
  • 再见,CSDN
  • MySQL总复习
  • 桌面平台层安全随手记录
  • 【Docker】 08-Dockerfile
  • 【二等奖方案】大规模金融图数据中异常风险行为模式挖掘赛题「Aries」解题思路
  • Github 下载指定文件夹(git sparse-checkout)
  • 蚂蚁集团SQLess 开源,与内部版有何区别?
  • An Efficient Memory-Augmented Transformer for Knowledge-Intensive NLP Tasks
  • Java项目中jar war pom包的区别
  • 整理mongodb文档:分页
  • 社区团购新玩法,生鲜蔬菜配货发货小程序商城
  • shell bash中设置命令set
  • 机器学习---预剪枝、后剪枝(REP、CCP、PEP、)
  • Python 爬虫—scrapy
  • ABB机器人20032转数计数器未更新故障报警处理方法
  • C# 记事本应用程序
  • 模型训练:优化人工智能和机器学习,完善DevOps工具的使用
  • WorldCoin 运营数据,业务安全分析
  • Java之Calender类的详细解析
  • uniapp 微信小程序 锚点跳转
  • 主成分分析笔记
  • android studio 的 adb配置
  • 【HTML5高级第一篇】Web存储 - cookie、localStorage、sessionStorage