Files
SimSwapPlus/clear_dataset.py
T
chenxuanhong 29d8914c0a update
2022-04-24 15:44:47 +08:00

31 lines
1.1 KiB
Python

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
#############################################################
# File: clear_dataset.py
# Created Date: Thursday March 24th 2022
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Sunday, 3rd April 2022 1:20:44 am
# Modified By: Chen Xuanhong
# Copyright (c) 2022 Shanghai Jiao Tong University
#############################################################
import os
import json
from utilities.json_config import readConfig, writeConfig
if __name__ == "__main__":
savePath = "./vggface2hq_failed.txt"
env_config = readConfig('env/env.json')
env_config = env_config["path"]
dataset_root = env_config["dataset_paths"]["vggface2_hq"]["images"]
# dataset_root = "G:/VGGFace2-HQ/newversion"
print(dataset_root)
with open(savePath,'r') as logf:
for line in logf:
img_path = os.path.join(dataset_root,line.replace("\n","")).replace("\\","/")
try:
os.rename(img_path,img_path+".deleted")
except Exception as e:
print(e)