init

Author: asahi417

.gitignore

@@ -1,3 +1,4 @@
 log_file
 wikiart_credential.json
-image-super-resolution
+image-super-resolution
+data

examples/generate_portrait_data.py

@@ -33,6 +33,7 @@ def get_export_path(_path):
 
 export_path = [get_export_path(i) for i in image_path]
 
+
 logging.info('Step 2: process for portrait dataset')
 processed_files = portrait_data_pipeline(image_path, export_path=export_path)
 logging.info('total {} images'.format(len(processed_files)))

wikiartcrawler/image_processors.py

@@ -198,6 +198,8 @@ def process_single_image(_image_path, _export_path):
                 pre_isr_image_path = new_file_path(_export_path, 'pre_isr', export_dir)
                 Image.fromarray(cv_img_out).save(pre_isr_image_path)
                 isr_model.predict(pre_isr_image_path, _export_path)
+                if not debug_mode:
+                    os.remove(pre_isr_image_path)
             return _export_path
         return None