Mask生成自己的txt文件----train和val两个

当你的标签数据集没有txt文件的时候，记得使用下面的的代码生成两个文件txt文件夹，分别为：train.txt和val.txt文件。代码如下：

data_root 目录是指Mask目录，不是原始的图片文件，是标签文件。

import mmcv
import os.path as osp
data_root = "/data3/datasets/Custom/Lab/Segmentation/"
ann_dir = "ann_png1"
split_dir = 'splits'
mmcv.mkdir_or_exist(osp.join(data_root, split_dir))
filename_list = [osp.splitext(filename)[0] for filename in mmcv.scandir(
    osp.join(data_root, ann_dir), suffix='.png')]
with open(osp.join(data_root, split_dir, 'train.txt'), 'w') as f:
  # select first 4/5 as train set
  train_length = int(len(filename_list)*4/5)
  f.writelines(line + '\n' for line in filename_list[:train_length])
with open(osp.join(data_root, split_dir, 'val.txt'), 'w') as f:
  # select last 1/5 as train set
  f.writelines(line + '\n' for line in filename_list[train_length:])

下面是参考别人博士的说法，参考网址为：
https://blog.csdn.net/weixin_44044411/article/details/118196847

data_root写自己的工作目录名称，
ann_dir写标签图片所在的目录，
split_dir则是在data_root下生成split txt文件保存的文件夹目录，其他的就不需要怎么改了。
如果你在data_root/split_dir/下成功找到了train.txt和val.txt文件，就没有问题了。