批量分割mask转xml检测框

import sys
import os
import cv2
import yaml
import copy
import xmltodict
import numpy as np
import skimage.io as io
import xml.etree.ElementTree as ET
from xml.dom.minidom import parse
from glob import glob

# single mask to xml&img
def SP(path):
    image = cv2.imread(path)
    locs = np.where(image==255)
    x0 =np.min(locs[1])
    x1 =np.max(locs[1])
    y0 =np.min(locs[0])
    y1 =np.max(locs[0])
    cv2.rectangle(image, (x0, y0), (x1, y1), (255, 0, 0), 2)
    cv2.imwrite('test.png', image)

def beatau(e,level=0):
    if len(e)>0:
        e.text='\n'+'\t'*(level+1)
        for child in e:
           beatau(child,level+1)
        child.tail=child.tail[:-1]
    e.tail='\n' + '\t'*level
def ToXml(name, path):
    root = ET.Element('annotation')#根节点
    erow1 = ET.Element('folder')#节点1
    erow1.text= "VOC"
    
    
    erow2 = ET.Element('filename')#节点2
    erow2.text= str(name)
    
    erow3 = ET.Element('size')#节点3
    erow31 = ET.Element('width')
    erow31.text = "512"
    erow32 = ET.Element('height')
    erow32.text = "512"
    erow33 = ET.Element('depth')
    erow33.text = "3" 
    erow3.append(erow31)
    erow3.append(erow32)
    erow3.append(erow33)
    root.append(erow1)
    root.append(erow2)
    root.append(erow3) 
    
    x0,y0,x1,y1 = XY(path)
    erow4 = ET.Element('object')
    erow41 = ET.Element('name')
    erow41.text = 'pco'
    erow42 = ET.Element('bndbox')
    erow4.append(erow41)
    erow4.append(erow42)
    erow421 = ET.Element('xmin')
    erow421.text = str(x0)
    erow422 = ET.Element('ymin')
    erow422.text = str(y0)
    erow423 = ET.Element('xmax')
    erow423.text = str(x1)
    erow424 = ET.Element('ymax')
    erow424.text = str(y1)
    erow42.append(erow421)
    erow42.append(erow422)
    erow42.append(erow423)
    erow42.append(erow424)
    root.append(erow4) 
    beatau(root)  
	return ET.ElementTree(root)    
    

for img in os.listdir('/home/hp/Desktop/shibie/pca_shibie/1'):
    path ='/home/hp/Desktop/shibie/pca_shibie/1' + '/' + img
    print(img)
    print(path)
    XML = ToXml(img, path)
    XML.write('./{}.xml'.format(img[:-4]))

nii to png

import numpy as np
import os  # 遍历文件夹
import nibabel as nib  # nii格式一般都会用到这个包
import imageio  # 转换成图像

def rename(imgfile):
    #rootDir = imgfile
    for root, dirs, files in os.walk(imgfile):
        path = root
        print(path)
        for f in files:
            os.rename(path + os.sep + f, path + os.sep + os.path.basename(path) + f)

def nii_to_image(niifile):
    filenames = os.listdir(filepath)  # 读取nii文件夹
    slice_trans = []

    for f in filenames:
        # 开始读取nii文件
        img_path = os.path.join(filepath, f)
        img = nib.load(img_path)  # 读取nii
        img_fdata = img.get_fdata()
        fname = f.replace('.nii*', '')  # 去掉nii的后缀名，替换成空的即为文件夹
        img_f_path = os.path.join(imgfile, fname)

        # 创建nii对应的图像的文件夹
        #if not os.path.exists(img_f_path):
            #os.mkdir(img_f_path)  # 新建文件夹

        # 开始转换为图像
        (x, y, z) = img.shape

        for i in range(z):  # z是图像的序列
            silce = img_fdata[:, :, i]  # 选择哪个方向的切片都可以
            # imageio.imwrite(os.path.join(img_f_path,'{}.png'.format(i)), silce)
            imageio.imwrite(os.path.join(imgfile, fname+'{}.png'.format(i)), silce)

            # 保存图像


if __name__ == '__main__':
    filepath = '/media/hp/4T/datasets/total001-300/imm001-200/'
    imgfile = '/media/hp/4T/datasets/total001-300/img_png/'
    nii_to_image(filepath)
    rename(imgfile)

二值化处理

import cv2
import os
from typing import Tuple
import numpy as np

import natsort

base_dir = "/media/hp/4T/datasets/PCA1-200/mask/"  # 要二值化的图片的原文件夹位置
output_dir = '/media/hp/4T/datasets/PCA1-200/mask2/'  # 二值化处理后存放的文件夹位置

if not os.path.exists(output_dir):
    os.makedirs(output_dir)

fileList = os.listdir(base_dir)
fileList = natsort.natsorted(fileList)


i = 1

for i in range(0,len(fileList)):
    path = os.path.join(base_dir, fileList[i])
    # print(path)
    img0 = cv2.imread(path,0)  # 读取格式为BGR
   
    
    mask = img0

    



    save_path = output_dir + path[36:]#保存至另一文件夹
    #save_path = os.path.join(output_dir, 'pca_' + str(s) + '_seg' + '.png'
    cv2.imwrite(save_path, mask)

    print( i+1 )
    print( save_path )