2021-06-13

最新推荐文章于 2024-10-06 23:55:17 发布

洛宾王

最新推荐文章于 2024-10-06 23:55:17 发布

阅读量121

点赞数

文章标签：深度学习

本文链接：https://blog.csdn.net/wanghrobin_1629/article/details/117867895

版权

@[TOC]语义分割数据集PNG文件整理为8bit格式的简单处理

语义分割数据集PNG文件整理为8bit格式的简单处理

PaddlePaddle深度学习的语义分割训练经常会用到PNG格式的图片文件。要求每像素数值在0-255之间，也就是8bit的灰度图。而不同研究机构和学习者分享的数据集格式不一。即使是PNG格式也一样有16bit、24bit等等，造成莫名其妙的训练失败。所以自己实验了一下，写了几行代码来刷文件格式。

为了方便，源文件和目标文件为两个分开的文件夹：Source和Target
##代码如下

import cv2
import os

SourcePath = "Source文件夹的路径"
TargetPath = "Target文件夹的路径"
filenames = os.listdir(SourcePath)

for f in filenames:
    source = os.path.join(SourcePath, f)
    image = cv2.imread(source)
    img_gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    target = os.path.join(TargetPath, f)
    cv2.imwrite(target, img_gray)