0

0

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

雪夜

雪夜

发布时间:2025-09-08 08:24:23

|

329人浏览过

|

来源于php中文网

原创

本人耗费数日时间,成功完成了windowsubuntu系统下caffe-ssd环境的搭建、脚本优化,并最终整合出一套高效且可自定义训练的完整项目包。该方案极大简化了训练流程,用户在配置好caffe-ssd环境后,无需再繁琐地编写或修改大量脚本,也不必进行复杂的数据格式转换,即可快速启动自己的数据集训练任务。虽然不敢称专家,但对整个流程已相当熟悉,所有核心脚本均为本人亲手编写。先展示一张在windows系统中训练自定义数据时的运行界面:

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

再来看一张Ubuntu系统下CPU版Caffe-SSD的训练截图:

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

两个平台的整体流程基本一致,但在脚本实现上略有差异:Windows主要依赖bat批处理文件,而Ubuntu则以Python和Shell脚本为主。接下来将分别介绍两个平台的使用方法。


Windows平台使用说明:

本方案的一大优势是高度自动化,只需将项目放置指定路径,几乎无需手动修改代码即可运行。

实验环境:

  • Windows 10 专业版 64位
  • Caffe-SSD CPU版本

操作步骤:

  1. 将源码解压至

    D:\caffe-ssd-microsoft
    目录下,并将
    myproj
    文件夹复制到该目录中,与主Caffe文件夹同级。

    windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

  2. 进入

    D:\caffe-ssd-microsoft\myproj\data\VOCdevkit\VOC2012
    目录:

    • 将自己的
      .jpg
      图像文件放入
      JPEGImages
      文件夹;
    • 对应的XML标注文件放入
      Annotations
      文件夹。
  3. 删除

    D:\caffe-ssd-microsoft\myproj\lmdb
    目录下的
    train_lmdb
    test_lmdb
    文件夹,避免后续生成LMDB时出错。

  4. 打开

    bat
    文件夹,其中包含一个
    xml2txt.py
    脚本,用于将XML标注转换为训练所需的文本列表。

    • 打开命令行窗口,切换至
      bat
      目录;
    • 执行命令:
      python xml2txt.py
    • 此时会在
      txt
      文件夹中生成
      train.txt
      test.txt
      ,格式为:
      图片名.xml名
    • 默认按80%训练集、20%测试集划分,如需调整比例,可修改脚本内参数。
  5. 继续在

    bat
    目录执行:
    python get_name_size.py

    • 生成
      test_name_size.txt
      文件,内容为:文件名(无后缀) 高度 宽度。
  6. 双击运行

    create_lmdb.bat
    ,自动完成训练与测试LMDB数据库的生成。
    ⚠️ 注意:请确保之前已删除旧的
    train_lmdb
    test_lmdb
    文件夹,否则会报错。

  7. 最后双击

    starttrain.bat
    ,即可启动模型训练流程。

    A1.art
    A1.art

    一个创新的AI艺术应用平台,旨在简化和普及艺术创作

    下载

Ubuntu平台使用说明:

由于Linux系统多采用绝对路径,因此需要根据实际用户名进行一次路径替换,其余部分基本无需更改。

实验环境:

  • Ubuntu 16.04
  • Caffe-SSD CPU版本

操作步骤:

  1. /home
    下新建
    caffe-ssd
    目录,将Caffe-SSD主程序放入其中,并将
    myproj
    文件夹与其置于同一层级。

    windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告
    windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

  2. 进入

    /home/fut/caffe-ssd/myproj/data/VOCdevkit/VOC2012

    • 图像放入
      JPEGImages
    • XML标注放入
      Annotations
  3. 清空

    /home/fut/caffe-ssd/myproj/lmdb
    中的
    train_lmdb
    test_lmdb
    文件夹。

  4. 打开

    script/xml2txt.py
    ,将其中的用户名(如
    fut
    )替换为当前系统的用户名,建议使用全局替换方式批量修改路径;

    • 切换到
      script
      目录,运行:
      python xml2txt.py
    • 生成
      train.txt
      test.txt
      ,格式同上,划分比例可在脚本中调整。
  5. 修改

    get_name_size.py
    中的用户名路径,保存后执行:
    python get_name_size.py

    • 生成
      test_name_size.txt
      ,记录测试集图像尺寸信息。
  6. 在终端执行:

    python train_lmdb_gen.py
    ,生成训练用LMDB数据;

    • 输出路径为
      lmdb/train_lmdb
  7. 执行:

    python test_lmdb_gen.py
    ,生成测试用LMDB数据;

    • 输出路径为
      lmdb/test_lmdb
  8. 最后运行:

    python ssd_pascal.py
    开始训练;

    • 该脚本中多数参数均可自定义,只需确保用户名路径正确即可;
    • 此配置适用于目标检测任务。
若在Caffe-SSD安装配置过程中遇到问题,可私信联系,提供有偿技术支持。

目前因设备限制暂未部署GPU版本,但基于现有结构,迁移至GPU环境难度不大,属于常规操作范畴。

相关文章

Windows激活工具
Windows激活工具

Windows激活工具是正版认证的激活工具,永久激活,一键解决windows许可证即将过期。可激活win7系统、win8.1系统、win10系统、win11系统。下载后先看完视频激活教程,再进行操作,100%激活成功。

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
pdf怎么转换成xml格式
pdf怎么转换成xml格式

将 pdf 转换为 xml 的方法:1. 使用在线转换器;2. 使用桌面软件(如 adobe acrobat、itext);3. 使用命令行工具(如 pdftoxml)。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

1946

2024.04.01

xml怎么变成word
xml怎么变成word

步骤:1. 导入 xml 文件;2. 选择 xml 结构;3. 映射 xml 元素到 word 元素;4. 生成 word 文档。提示:确保 xml 文件结构良好,并预览 word 文档以验证转换是否成功。想了解更多xml的相关内容,可以阅读本专题下面的文章。

2119

2024.08.01

xml是什么格式的文件
xml是什么格式的文件

xml是一种纯文本格式的文件。xml指的是可扩展标记语言,标准通用标记语言的子集,是一种用于标记电子文件使其具有结构性的标记语言。想了解更多相关的内容,可阅读本专题下面的相关文章。

1168

2024.11.28

windows查看端口占用情况
windows查看端口占用情况

Windows端口可以认为是计算机与外界通讯交流的出入口。逻辑意义上的端口一般是指TCP/IP协议中的端口,端口号的范围从0到65535,比如用于浏览网页服务的80端口,用于FTP服务的21端口等等。怎么查看windows端口占用情况呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1496

2023.07.26

查看端口占用情况windows
查看端口占用情况windows

端口占用是指与端口关联的软件占用端口而使得其他应用程序无法使用这些端口,端口占用问题是计算机系统编程领域的一个常见问题,端口占用的根本原因可能是操作系统的一些错误,服务器也可能会出现端口占用问题。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1170

2023.07.27

windows照片无法显示
windows照片无法显示

当我们尝试打开一张图片时,可能会出现一个错误提示,提示说"Windows照片查看器无法显示此图片,因为计算机上的可用内存不足",本专题为大家提供windows照片无法显示相关的文章,帮助大家解决该问题。

835

2023.08.01

windows查看端口被占用的情况
windows查看端口被占用的情况

windows查看端口被占用的情况的方法:1、使用Windows自带的资源监视器;2、使用命令提示符查看端口信息;3、使用任务管理器查看占用端口的进程。本专题为大家提供windows查看端口被占用的情况的相关的文章、下载、课程内容,供大家免费下载体验。

463

2023.08.02

windows无法访问共享电脑
windows无法访问共享电脑

在现代社会中,共享电脑是办公室和家庭的重要组成部分。然而,有时我们可能会遇到Windows无法访问共享电脑的问题。这个问题可能会导致数据无法共享,影响工作和生活的正常进行。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2361

2023.08.08

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

3

2026.03.11

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
PostgreSQL 教程
PostgreSQL 教程

共48课时 | 10.5万人学习

Git 教程
Git 教程

共21课时 | 4.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号