Parcourir la source

Merge branch 'release/v0.1.2'

Jérôme BUISINE il y a 5 ans
Parent
commit
97f222f99c
1 fichiers modifiés avec 11 ajouts et 1 suppressions
  1. 11 1
      utils/data.py

+ 11 - 1
utils/data.py

@@ -34,10 +34,20 @@ def get_renderer_scenes_names(renderer_name):
 
 def get_scene_image_quality(img_path):
 
+    # if path getting last element (image name) and extract quality
     img_postfix = img_path.split('/')[-1].split(scene_image_quality_separator)[-1]
     img_quality = img_postfix.replace(scene_image_extension, '')
 
-    return img_quality
+    return int(img_quality)
+
+
+def get_scene_image_prefix(img_path):
+
+    # if path getting last element (image name) and extract prefix
+    img_prefix = img_path.split('/')[-1].split(scene_image_quality_separator)[0]
+
+    return img_prefix
+
 
 def augmented_data_image(block, output_folder, prefix_image_name):