mask_rcnn 训练自己数据——收集资料

mask_rcnn 训练自己数据

欢迎使用Markdown编辑器

主要参考及工具

基于Mask RCNN开源项目:https://github.com/matterport/Mask_RCNN
图片标记工具基于开源项目:https://github.com/wkentaro/labelme

有关mask-rcnn和Faster RCNN算法可以参考:
https://blog.csdn.net/linolzhang/article/details/71774168
https://blog.csdn.net/lk123400/article/details/54343550/

新的改变

准备训练数据集

  1. 原图数据集(训练数据,测试数据)
  2. 用Labelme对数据集进行 标注,得到 .json 格式的数据, 在标注完保存后会为每张RGB图生成一个.json的标注文件,最后得到的所有的标注文件如下
    在这里插入图片描述
  3. Labelme数据集到COCO数据集的转化。 .json 数据格式转换,生成 img, info.yaml, label ,label_names , label_viz五个文件,。(这个是处理.json文件后产生的数据,使用方法为labelme_json_to_dataset+空格+文件名称.json,这个前提是labelme要准确安装并激活。但是这样会产生一个问题,对多幅图像这样处理,太麻烦,在这里提供一个工具,可以直接在.json文件目录下转换所有的json文件,链接:json数据转换工具
    转化工具 :https://download.csdn.net/download/u014525760/10348022
  4. 转化label.png为可用格式
    由于labelme生成的掩码标签 label.png为16位存储,opencv默认读取8位,需要将16位转8位,可通过C++程序转化,代码参考这篇博:http://blog.csdn.net/l297969586/article/details/79154150
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对于训练自己的数据集,你可以按照以下步骤进行: 1. 数据准备:收集并标注自己的数据集。标注的方式取决于你的任务类型,对于目标检测任务,你可以使用标注工具如LabelImg来为图像中的对象标注边界框和类别。 2. 配置环境:安装必要的依赖库和框架,如Python、OpenCV、PyTorch等。确保正确配置CUDA和cuDNN,以便在GPU上进行训练。 3. 模型选择:根据你的任务需求选择合适的模型,如Mask R-CNN。可以使用已经预训练好的模型作为基础网络,然后进行微调。 4. 数据预处理:将数据集划分为训练集和验证集,并进行数据预处理,如图像大小调整、颜色归一化、数据增强等。 5. 构建数据加载器:使用相应的库构建数据加载器,以便将数据输入到模型中进行训练。 6. 模型训练:使用训练集对模型进行训练。在训练过程中,可以使用学习率衰减、权重衰减等技巧来提高模型性能。 7. 模型评估:使用验证集评估模型性能,可以计算精确度、召回率、F1分数等指标。 8. 模型调优:根据评估结果,对模型进行调优,如调整超参数、增加训练数据等。 9. 模型保存与部署:将训练好的模型保存为文件,并根据需要进行部署,可以使用模型进行预测或在其他应用中使用。 以上是大致的步骤,具体实现细节会根据你的任务和数据集的特点而有所不同。希望对你有所帮助!如有更多问题,请继续提问。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值