Spaces:

gyrojeff
/

YuzuMarker.FontDetection

Running

gyrojeff commited on Mar 25, 2023

Commit

daa52ce

•

1 Parent(s): c681b80

fix: typo

Files changed (1) hide show

detector/data.py CHANGED Viewed

@@ -82,8 +82,8 @@ class FontDataModule(LightningDataModule):
         self,
         config_path: str = "configs/font.yml",
         train_path: str = "./dataset/font_img/train",
-        val_path: str = "./dataset/font_img/train",
-        test_path: str = "./dataset/font_img/train",
         train_shuffle: bool = True,
         val_shuffle: bool = False,
         test_shuffle: bool = False,

         self,
         config_path: str = "configs/font.yml",
         train_path: str = "./dataset/font_img/train",
+        val_path: str = "./dataset/font_img/val",
+        test_path: str = "./dataset/font_img/test",
         train_shuffle: bool = True,
         val_shuffle: bool = False,
         test_shuffle: bool = False,