54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
-------------------------------------------------
|
|||
|
|
Project Name: unet
|
|||
|
|
File Name: gen_split.py
|
|||
|
|
Author: chenming
|
|||
|
|
Create Date: 2022/2/6
|
|||
|
|
Description：
|
|||
|
|
-------------------------------------------------
|
|||
|
|
"""
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
-------------------------------------------------
|
|||
|
|
Project Name: sd_segmentation
|
|||
|
|
File Name: gen_split.py.py
|
|||
|
|
Author: chenming
|
|||
|
|
Create Date: 2022/2/6
|
|||
|
|
Description：
|
|||
|
|
-------------------------------------------------
|
|||
|
|
"""
|
|||
|
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: UTF-8 -*-
|
|||
|
|
'''
|
|||
|
|
@Project £ºai_train_code
|
|||
|
|
@File £ºgen_txt.py
|
|||
|
|
@Author £ºChenmingSong
|
|||
|
|
@Date £º2021/12/31 18:42
|
|||
|
|
@Description£º
|
|||
|
|
'''
|
|||
|
|
import os
|
|||
|
|
import random
|
|||
|
|
import numpy as np
|
|||
|
|
|
|||
|
|
# annotations_foder_path = "E:/biye/gogogo/english/detection_yolov5/official/data/origin/VOC2012/Annotations"
|
|||
|
|
annotations_foder_path = "/scm/data/seg/xianyu/skin_seg/data/ISBI2016_ISIC_Part1_Training_Data"
|
|||
|
|
names = os.listdir(annotations_foder_path)
|
|||
|
|
real_names = [name.split(".")[0] for name in names]
|
|||
|
|
print(real_names)
|
|||
|
|
random.shuffle(real_names)
|
|||
|
|
print(real_names)
|
|||
|
|
length = len(real_names)
|
|||
|
|
split_point = int(length * 0.3)
|
|||
|
|
|
|||
|
|
val_names = real_names[:split_point]
|
|||
|
|
train_names = real_names[split_point:]
|
|||
|
|
|
|||
|
|
# ¿ªÊ¼Éú³ÉÎļþ
|
|||
|
|
np.savetxt('val.txt', np.array(val_names), fmt="%s", delimiter="\n")
|
|||
|
|
np.savetxt('test.txt', np.array(val_names), fmt="%s", delimiter="\n")
|
|||
|
|
np.savetxt('train.txt', np.array(train_names), fmt="%s", delimiter="\n")
|
|||
|
|
# print("txtÎļþÉú³ÉÍê±Ï£¬Çë·ÅÔÚVOC2012µÄImageSets/MainµÄĿ¼ÏÂ")
|
|||
|
|
|
|||
|
|
np.savetxt('bbbbb.txt', np.array(real_names), fmt="%s", delimiter="\n")
|