当前位置: 代码迷 >> 综合 >> 【教程】mask-rcnn pytorch版本训练自己的数据集-猫狗识别
  详细解决方案

【教程】mask-rcnn pytorch版本训练自己的数据集-猫狗识别

热度:93   发布时间:2023-10-19 17:57:46.0

整个项目配置过程和教程讲解过程的录制视频,可搜索b站或西瓜视频up:杭漂一族小张。

需要demo或有demo需求的可以联系扣扣2014453702

这个专栏更新各种AI,以及各种有趣的教程,有兴趣的小伙伴可以订阅一下。

首先是数据集的准备,这里以猫狗识别为例子

【教程】mask-rcnn pytorch版本训练自己的数据集-猫狗识别

使用labelme对图片进行打标签,生成json格式的图片,而且labelme的版本尽量新一点,老版本 比如3.16.7版本 对后面生成SegmentationClassPNG 和 SegmentationObjectPNG会有问题。

直接在命令行输入labelme即可

labelme

【教程】mask-rcnn pytorch版本训练自己的数据集-猫狗识别

 在使用labelme打上标签之后,就会在本地生成json格式

【教程】mask-rcnn pytorch版本训练自己的数据集-猫狗识别

  相关解决方案