图像处理数据集txt文件自定义

对Cityscapes和CamVid数据集的索引txt文件修改

import numpy as np
import os
import torch.nn as nn
from PIL import Image
 
 
def default_loader(path):
    return Image.open(path).convert('RGB')

 
def TxtData(label_path, train_path, txt_path):
    txt_name = 'test.txt'
    txt_path = os.path.join(txt_path, txt_name)
    label = os.listdir(label_path)
    train = os.listdir(train_path)
    ‘’‘label.sort(key=lambda x: (int(x.split('_')[0]), int(x.split('.')[0].split('_')[1])))#对文件名称进行排序
    train.sort(key=lambda x: (int(x.split('_')[0]), int(x.split('.')[0].split('_')[1])))’‘’
    file = open(txt_path, 'w')
 
    if len(label) != len(train):
        print('the number of traing data varies between the two files')
 
    for i,j in zip(label, train):
        path = os.path.join(train_path,j)+ ' ' +os.path.join(label_path,i) + '\n'
        file.write(path)
 
 
if __name__ == '__main__':
    data_label = 'trainannot' #标签地址
    data_train = 'train' #原图地址
    txt_path = ' ' #生成的txt地址
    TxtData(data_label, data_train, txt_path)