More logging

Author: Oxid15

datagen/generator.py

@@ -190,7 +190,7 @@ def _rendering_loop(self, writer: Union[csv.DictWriter, None]) -> None:
                     logging.info("Finished at max images")
                     break
 
-            logging.info(f"Generated for {i + 1} input images")
+            logging.info(f"Generated images for {i + 1} inputs")
 
     def run(self) -> None:
         os.makedirs(os.path.join(self.dst, "images"))
@@ -202,6 +202,7 @@ def run(self) -> None:
             csv_file = open(os.path.join(self.dst, "map.csv"), "w")
             writer = csv.DictWriter(csv_file, ["base_img", "query_img"])
 
+        logging.info("Starting generation process")
         try:
             self._rendering_loop(writer)
         except Exception as e:
@@ -213,4 +214,7 @@ def run(self) -> None:
                 logging.info("Map file closed")
 
         if self.generate_meta:
+            logging.info("Writing meta")
             self._write_meta()
+
+        logging.info("Done!")