一、准备自己的数据集

在Pytorch-UNet-master\data下创建 imgs和masks两个文件夹

其中imgs存放rgb（.jpg格式)图像， masks存放标签（.png）图像

此时我手中已经有如下的jpg图像和png图像

但直接使用这样的png图像训练会报错

因为目标（白色）像素值是255

所以我写了一个脚本可以批量把图像转化为像素值为0，1的图像

import os
import cv2
from PIL import Imagedef binarization():# 获取目录下所有图片名filename = os.listdir(r"E:\..")print(filename)# os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。base_dir = r"E:\.." # inputnew_dir  = r"E:\.."  # outputfor img in filename:name = imgpath1 = os.path.join(base_dir ,img)img = cv2.imread(path1)# print(img)Grayimg = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)ret, thresh = cv2.threshold(Grayimg, 170, 1 ,cv2.THRESH_BINARY)cv2.imwrite('img.png', thresh)image = Image.open('img.png')# 有需要可对图像进行大小调整# image = image.resize((350, 350),Image.ANTIALIAS)path = os.path.join(new_dir ,name)image.save(path)binarization()

之后生成的标签图为

此时背景的像素值为0，目标像素为1

最后把调整过后的png放入masks文件夹，jpg放入imgs文件夹

python train.py

即可开始训练

Unet（pytorch）——制作自己的数据集（二分类）相关推荐

（！详解 Pytorch实战：①）kaggle猫狗数据集二分类：加载（集成/自定义）数据集
这系列的文章是我对Pytorch入门之后的一个总结,特别是对数据集生成加载这一块加强学习另外,这里有一些比较常用的数据集,大家可以进行下载: 需要注意的是,本篇文章使用的PyTorch的版本是v0. ...
pytorch创建自己的数据集（分类任务）
pytorch创建自己的数据集(分类任务) 转载于:https://www.cnblogs.com/cititude/p/11615158.html
ML之FE之FS：特征工程/数据预处理—特征选择之利用过滤式filter、包装式wrapper、嵌入式Embedded方法(RF/SF)进行特征选择(mushroom蘑菇数据集二分类预测)最全案例应用
ML之FE之FS:特征工程/数据预处理-特征选择之利用过滤式filter.包装式wrapper.嵌入式Embedded方法(RF/SF)进行特征选择(mushroom蘑菇数据集二分类预测)案例应用利 ...
鸢尾花数据集、月亮数据集二分类可视化分析
鸢尾花数据集.月亮数据集二分类可视化分析目录鸢尾花数据集.月亮数据集二分类可视化分析一.线性LDA算法 1.LDA算法 2.鸢尾花数据集 2.月亮数据集二.K-Means算法 1.K-Mean ...
《pytorch》对CIFAR数据集的分类
今天复习了下pytorch,又看了下它的一个实例,复习了下对CIFAR数据集的分类. 学习链接:https://zhuanlan.zhihu.com/p/39667289 直接上源码,注释均在其中 i ...
PyTorch做神经网络MINIST数据集多分类
PyTorch是Facebook的AI研究团队发布的python工具包,主要用于深度学习.这篇文章我们一起学习一下如何使用PyTorch搭建神经网络训练分类模型,这里我们用的数据集是Yann LeCu ...
LESSON 10.110.210.3 SSE与二分类交叉熵损失函数二分类交叉熵损失函数的pytorch实现多分类交叉熵损失函数
在之前的课程中,我们已经完成了从0建立深层神经网络,并完成正向传播的全过程.本节课开始,我们将以分类深层神经网络为例,为大家展示神经网络的学习和训练过程.在介绍PyTorch的基本工具AutoGrad ...
Pytorch之模型微调（Finetune）——用Resnet18进行蚂蚁蜜蜂二分类为例
Pytorch之模型微调(Finetune)--手写数字集为例文章目录 Pytorch之模型微调(Finetune)--手写数字集为例前言一.Transfer Learning and Mode ...
pytorch_lesson10 二分类交叉熵损失函数及调用+多分类交叉熵损失函数及调用
注:仅仅是学习记录笔记,搬运了学习课程的ppt内容,本意不是抄袭!望大家不要误解!纯属学习记录笔记!!!!!! 文章目录一.机器学习中的优化思想二.回归:误差平方和SSE 三.二分类交叉熵损失函数 ...
机器学习：二分类到多分类-ovr,ovo,mvm,sofmax
一般情况下,我们都认为逻辑回归(LR)用来解决二分类问题,BCE损失函数也是用来解决二分类问题的,那逻辑回归以及BCE损失函数能否用来做多分类任务呢,答案是肯定的.有两种方法一.将多分类任务拆解成多 ...

Unet（pytorch）——制作自己的数据集（二分类）

一、准备自己的数据集

Unet（pytorch）——制作自己的数据集（二分类）相关推荐

最新文章

热门文章