# 生成训练数据的路径文本
# 注意:1.生成训练还是测试数据的路径; 2.txt往后追加路径; 3.共4个形状,需要改动shape1,2,3,4
def generate_path():
data_path = '.\\Dataset\shape4\\train_data\\Output_rgb'
txt_path = '.\\Dataset\\train_path.txt'
file = open(txt_path, mode='a') # 向txt文件中追加
for (path, dirnames, filenames) in os.walk(data_path):
for filename in filenames:
if filename.endswith('png'):
rgb_path = path + '\\' + filename
# print('当前路径 %s' % rgb_path)
example = int(rgb_path.split('\\')[5][3:]) #当前example
if example <= 49:
print(example)
#print(rgb_path)
depth_path = rgb_path.replace('rgb', 'depth')
#print(depth_path)
line = rgb_path + ' ' + depth_path + '\n'
file.write(line)
#file.write('test\n')
file.close()
生成的文本格式
def copy_img():
input_dir = 'E:\\m_DORN\DataSet\\shape1\\test_data\\Output_rgb'
output_dir = 'E:\\m_DORN\\DataSet\\test_data\\rgb'
# 新建输出文件夹
if not os.path.exists(output_dir):
os.mkdir(output_dir)
for (path, dirnames, filenames) in os.walk(input_dir):
for filename in filenames:
if filename.endswith('png'):
img_path = path + '/' + filename
print('Being process img %s' % filename)
shutil.copy(img_path, output_dir)