图像分割
文章平均质量分 91
EdenGabriel
no why,just do.
展开
-
2.1【Pytorch版(torch-cuda版 ) Mask-RCNN 训练自己的数据集】
写在前面:由于课程试验要求,需要基于pytorch实现maskrcnn,so最近又跑了一下pytorch版的maskrcnn,官方已经给出了详细的教程,虽然说支持cpu推理,但是不支持cpu训练啊,奈何手头上只有一个cpu本,也没有nvida显卡,只有intel的集显,so整理一波本次训练maskrcnn的过程。这篇blog是按照常规的torch版的maskrcnn的训练教程写的,理论上,所依赖的是torch-cuda版本,后续还有一篇是基于torch-cpu版本的。因为在试验过程中先踩了这部分的坑,.原创 2020-05-15 22:27:43 · 9936 阅读 · 11 评论 -
2.1【Pytorch版(torch-CPU版 ) Mask-RCNN 训练自己的数据集】(无需安装torch-cuda,在无nvida显卡的电脑下跑通)
写在前面:由于课程试验要求,需要基于pytorch实现maskrcnn,so最近又跑了一下pytorch版的maskrcnn,官方已经给出了详细的教程,虽然说支持cpu推理,但是不支持cpu训练啊,奈何手头上只有一个cpu本,也没有nvida显卡,只有intel的集显,so整理一波本次训练maskrcnn的过程。环境:Ubuntu16.04torch == 1.5.0+cputorchvision == 0.6.0+cpu这里要注意,torch版本>=0.3.0即可,使用的torch和t.原创 2020-05-15 22:26:55 · 5065 阅读 · 1 评论 -
1.3 【Mask-RCNN训练自己的数据集】---- Part Three: 用自己训练的模型进行测试(全部流程总结+部分释义)
三.模型测试import osimport syssys.path.remove('/opt/ros/kinetic/lib/python2.7/dist-packages')import randomimport mathimport numpy as npimport skimage.ioimport matplotlibimport matplotlib.pyplot as...原创 2020-01-09 22:37:02 · 736 阅读 · 0 评论 -
1.2【Mask-RCNN训练自己的数据集】---- Part Two: 模型训练(全部流程总结+部分释义)
二.编写模型训练代码打开MaskRCNN文件夹下的samples文件夹,存在以下四个文件夹:下述模型训练代码将对shapes文件夹下的train_shapes.ipnyb 进行修改(这里是jupyter的文本格式,笔者不太习惯使用这种格式,所以将其转换为py文件。)2.1 下载coco预训练权重文件MODEL_DIR = os.path.join(ROOT_DIR, "logs")it...原创 2020-01-09 22:36:22 · 2537 阅读 · 10 评论 -
1.1【Mask-RCNN训练自己的数据集】---- Part One:制作数据集(全部流程总结+部分释义)
写在前面该实现基于Tensorflow、Keras框架,在Ubuntu16.04下运行。官方github MaskRCNN demo一.制作数据集1.下载安装LabelmeLabelme用来制作图像分割数据集,在安装Labelme之前需要安装下载其所需要的一些依赖库,具体的安装方式见这篇博客Labelme安装Labelme github链接由于笔者之前已经安装过所需要的依赖库,...原创 2020-01-09 22:35:13 · 3036 阅读 · 5 评论