106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
import glob
|
|
import os
|
|
import shutil
|
|
import cv2
|
|
|
|
def merge_images(input_folder, output_folder):
|
|
if not os.path.exists(output_folder):
|
|
os.mkdir(output_folder)
|
|
|
|
count = 0
|
|
for root, _, filenames in os.walk(input_folder):
|
|
for fn in filenames:
|
|
if fn.endswith('.jpg'):
|
|
count += 1
|
|
shutil.copy(os.path.join(root, fn), os.path.join(output_folder, 'c_{:010d}.jpg'.format(count)))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
merge_images('p/beach', 'output')
|
|
|
|
for f in glob.glob(os.path.join('/home/logic/test/data/1xx_Beach', 'c_*.jpg')):
|
|
os.remove(f)
|
|
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/1xx_Beach')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('beach done')
|
|
|
|
merge_images('p/building', 'output')
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/2xx_Building')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('building done')
|
|
|
|
merge_images('p/bus', 'output')
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/3xx_Bus')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('bus done')
|
|
|
|
merge_images('p/dinosaur', 'output')
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/4xx_Dinosaur')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('dinosaur done')
|
|
|
|
merge_images('p/elephant', 'output')
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/5xx_Elephant')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('elephant done')
|
|
|
|
merge_images('p/horse', 'output')
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/7xx_Horse')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('horse done')
|
|
|
|
merge_images('p/mountain', 'output')
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/8xx_Mountain')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('mountain done')
|
|
|
|
merge_images('p/dish', 'output')
|
|
for f in glob.glob('output/*.jpg'):
|
|
img = cv2.imread(f)
|
|
if img is None:
|
|
continue
|
|
shutil.copy(f, '/home/logic/test/data/9xx_Dish')
|
|
for f in glob.glob('output/*.jpg'):
|
|
os.remove(f)
|
|
print('dish done')
|
|
|
|
print('done')
|
|
|