python対応フォルダに画像をコピー
866 ワード
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Mon Dec 10 14:10:59 2018
@author: yxh
"""
import numpy as np
import os
import shutil
import sys
newpath = '/home/yxh/caffe/examples/Vegetable/20181210TEST-11/'
f = open("/home/yxh/caffe/examples/Vegetable/20181210TEST-11/test.txt")
files = f.readlines()
for pic in files:
picname = pic.split(' ')[0]
picpath = '/home/yxh/caffe/examples/Vegetable/VegetableImage11' + picname
dir_name = picname.split('/')[1]
pic_name = picname.split('/')[2]
if not os.path.exists(dir_name):
os.makedirs(dir_name)
pic_name, pic_extend = os.path.splitext(pic_name)
new_path = os.path.join(newpath, dir_name) + '/' + pic_name + '.jpg'
shutil.copy(picpath, new_path)