Spaces:

ymcmy
/

AI_Generated_Image_Detector

Running

ymcmy commited on Jun 24, 2024

Commit

845a877

verified ·

1 Parent(s): ce88d40

Update preprocess.py

Files changed (1) hide show

preprocess.py CHANGED Viewed

@@ -116,7 +116,7 @@ class2_dirs = [
     "/home/AI_detection_dataset/Real_AI_SD_LD_Dataset/train/fake/",
     "/home/AI_detection_dataset/Real_AI_SD_LD_Dataset/test/fake/"
     ] #fake 1
-output_dir = "/content/drive/MyDrive/joblibsaves"
 file_paths_pickle_save_dir='/content/drive/MyDrive/aigc_file_paths.pkl'
 patch_num = 128
 N = 256

     "/home/AI_detection_dataset/Real_AI_SD_LD_Dataset/train/fake/",
     "/home/AI_detection_dataset/Real_AI_SD_LD_Dataset/test/fake/"
     ] #fake 1
+output_dir = "/content/drive/MyDrive/h5saves"
 file_paths_pickle_save_dir='/content/drive/MyDrive/aigc_file_paths.pkl'
 patch_num = 128
 N = 256