Add conversion to FEN

rpi/training/utils/sort_labels.py

@@ -2,15 +2,12 @@ import os
 import shutil
 
 
-def copy_images(src, dest):
-    image_extensions = [".jpg", ".jpeg", ".png", ".bmp", ".gif"]
+def copy_files(src, dest):
 
     for filename in os.listdir(src):
-        if any(filename.lower().endswith(ext) for ext in image_extensions):
-            src_path = os.path.join(src, filename)
-            dst_path = os.path.join(dest, filename)
-            shutil.copy2(src_path, dst_path)
-
+        src_path = os.path.join(src, filename)
+        dst_path = os.path.join(dest, filename)
+        shutil.copy2(src_path, dst_path)
 
 def remap_labels(src, dest):
     count = 0
@@ -41,8 +38,8 @@ def remap_labels(src, dest):
 
 if __name__ == "__main__":
 
-    src_dir = "../datasets/pieces/visualizan/"
-    dest_dir = "../datasets/pieces/unified/"
+    src_dir = "../datasets/edges/misis2025_cv_chess.v1i.yolov11/"
+    dest_dir = "../datasets/edges/"
 
     reference_classes = [
         'w_pawn', 'w_knight', 'w_bishop', 'w_rook', 'w_queen', 'w_king',
@@ -67,5 +64,6 @@ if __name__ == "__main__":
         dst_image_folder = os.path.normpath(os.path.join(dest_full_path, "images"))
         dst_labels_folder = os.path.normpath(os.path.join(dest_full_path, "labels"))
 
-        copy_images(src_image_folder, dst_image_folder)
-        remap_labels(src_labels_folder, dst_labels_folder)
+        copy_files(src_image_folder, dst_image_folder)
+        copy_files(src_labels_folder, dst_labels_folder)
+        #remap_labels(src_labels_folder, dst_labels_folder)