update,

vinniesniper-54816/task1/_lab/003-crawler-mountain/import_img copy.py

@@ -0,0 +1,105 @@
+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')
+