智能秤分析

tools/getpairs.py

@@ -9,8 +9,23 @@ import logging
 class PairGenerator:
     """Generate positive and negative image pairs for contrastive learning."""
 
-    def __init__(self):
+    def __init__(self, original_path):
         self._setup_logging()
+        self.original_path = original_path
+        self._delete_space()
+
+    def _delete_space(self):  # 删除图片文件名中的空格
+        print(self.original_path)
+        for root, dirs, files in os.walk(self.original_path):
+            for file_name in files:
+                if file_name.endswith('.jpg' or '.png'):
+                    n_file_name = file_name.replace(' ', '')
+                    os.rename(os.path.join(root, file_name), os.path.join(root, n_file_name))
+                if 'rotate' in file_name:
+                    os.remove(os.path.join(root, file_name))
+            for dir_name in dirs:
+                n_dir_name = dir_name.replace(' ', '')
+                os.rename(os.path.join(root, dir_name), os.path.join(root, n_dir_name))
 
     def _setup_logging(self):
         """Configure logging settings."""
@@ -188,11 +203,11 @@ class PairGenerator:
 
 
 if __name__ == "__main__":
-    original_path = '/home/lc/data_center/scatter/v4/val'
+    original_path = '/home/lc/data_center/contrast_data/v1/extra'
     parent_dir = str(Path(original_path).parent)
-    generator = PairGenerator()
+    generator = PairGenerator(original_path)
 
     # Example usage:
     pairs = generator.get_pairs(original_path,
-                                output_txt=os.sep.join([parent_dir, 'cross_same.txt']))  # Individual pairs
+                                output_txt=os.sep.join([parent_dir, 'extra_cross_same.txt']))  # Individual pairs
     # groups = generator.get_group_pairs('val')  # Group pairs