DeepFeatureConsistentVAE/clean.py at main · ImportPaddle/DeepFeatureConsistentVAE · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
# -*-coding:utf-8-*-
import os

SAVELIST = ['train_retrieval.py', 'clean.py', '__init__.py']
version = '7'


def deleteNouese(path):  # 遍历指定文件夹中所有文件，检查图像大小，长高小于300的删除,不是图像的文件也删除
    for root, dirs, files in os.walk(path):
        for file in files:
            if file in SAVELIST:
                continue
            name = file.rsplit('.', 1)[0]
            postfix = file.rsplit('.', 1)[-1]
            # print(file.rsplit('.',1))
            if postfix == 'py':
                pycPath = os.path.join(root, '__pycache__', name + '.cpython-3{}.pyc'.format(version))
                flag = os.path.exists(pycPath)
                # print(name+'.cpython-38.pyc')
                if flag:
                    print('save ', os.path.join(root, '__pycache__', name + '.cpython-3{}.pyc').format(version))
                    pass
                else:
                    pypath = os.path.join(root, file)
                    print('delete {}'.format(pypath))
                    # print('delete {}'.format(pycPath))

                    # try:
                    os.remove(pypath)
                    # os.remove(pycPath)
                    # except:
                    #     pass


def deletepyc(path):
    for root, dirs, files in os.walk(path):
        for file in files:
            aa1 = os.path.join(root, file)
            # print(aa1)
            if aa1.split('.')[-1] == 'pyc':
                print(aa1)
                os.remove(aa1)  # 删除文件


def main():
    path = "./torch_dfcvae"
    # deleteNouese(path)
    deletepyc(path)


if __name__ == "__main__":
    main()