获取人脸数据集

2024-11-28 14:47:32 +08:00
parent 58baaf68e1
commit 1ef65ff1a2
7 changed files with 113 additions and 7 deletions
@@ -16,5 +16,5 @@ def save_img_subset(data, save_path, num_samples):
        img.save(os.path.join(save_path, f"{i}-label-{label}.png"))

 # 保存前 600 张训练集图片和前 100 张测试集图片
-save_img_subset(train_data, './DataImages-Train', 6000)
-save_img_subset(test_data, './DataImages-Test', 1000)
+save_img_subset(train_data, './DataImages-Train', 60000)
+save_img_subset(test_data, './DataImages-Test', 10000)