Parcourir la source

Merge branch 'release/v0.3.7'

Jérôme BUISINE il y a 4 ans
Parent
commit
6190e5c234

Fichier diff supprimé car celui-ci est trop grand
+ 271 - 0
analysis/.ipynb_checkpoints/data_augmentation_analysis-checkpoint.ipynb


Fichier diff supprimé car celui-ci est trop grand
+ 271 - 0
analysis/data_augmentation_analysis.ipynb


+ 6 - 4
generate/generate_reconstructed_data.py

@@ -129,13 +129,15 @@ def generate_data(transformation, _scenes):
 
                     # Data augmentation!
                     rotations = [0, 90, 180, 270]
-                    img_flip_labels = ['original', 'horizontal', 'vertical', 'both']
+                    #img_flip_labels = ['original', 'horizontal', 'vertical', 'both']
+                    img_flip_labels = ['original', 'horizontal']
 
                     horizontal_img = output_block_img.transpose(Image.FLIP_LEFT_RIGHT)
-                    vertical_img = output_block_img.transpose(Image.FLIP_TOP_BOTTOM)
-                    both_img = output_block_img.transpose(Image.TRANSPOSE)
+                    #vertical_img = output_block_img.transpose(Image.FLIP_TOP_BOTTOM)
+                    #both_img = output_block_img.transpose(Image.TRANSPOSE)
 
-                    flip_images = [output_block_img, horizontal_img, vertical_img, both_img]
+                    #flip_images = [output_block_img, horizontal_img, vertical_img, both_img]
+                    flip_images = [output_block_img, horizontal_img]
 
                     # rotate and flip image to increase dataset size
                     for id, flip in enumerate(flip_images):