python 删除比较早的文件

若文件中的图片数目超过了1000张,删除多余的图片

删除文件夹中的图片

'''根据图片文件夹大小定时删除文件夹部分比较旧的数据'''

import os
def fileremove(filename):
    if os.path.exists(filename):
        os.remove(filename)
        print('remove file: %s' % filename)
    else:
        print('no such file: %s' % filename)

def check_image(ITEMS,image_num,FILE_DIR,max_image_saved):
    create_time = []
    file_name = []
    for names in ITEMS:
        if names.endswith(".jpg"):
            filename = os.path.join(FILE_DIR, names)
            file_name.append(filename)
            create_time.append(os.path.getctime(filename))  # 获取文件的修改时间
    # 将两个list转换为dict
    dit = dict(zip(create_time, file_name))
    # 根据dit的key对dit进行排序（变为list）
    dit = sorted(dit.items(), key=lambda d: d[-2], reverse=False)
    for i in range(image_num - max_image_saved):  # 删除文件个数
        file_location = dit[i][1]
        fileremove(file_location)

'''计算文件夹容量 单位为MB'''
if __name__ == '__main__':
    FILE_DIR = './images_out'
    max_image_saved=5
    ITEMS = os.listdir(FILE_DIR)  # FILE_DIR中的所有文件
    print('lent',len(ITEMS))
    image_num=len(ITEMS)
    if image_num>max_image_saved:
        check_image(ITEMS,image_num,FILE_DIR,max_image_saved)

删除文件夹中的训练模型

'''删除文件夹中比较旧的模型'''

import os
def fileremove(filename):
    if os.path.exists(filename):
        os.remove(filename)
        print('remove file: %s' % filename)
    else:
        print('no such file: %s' % filename)

def check_model(ITEMS,image_num,FILE_DIR,max_image_saved):
    create_time = []
    file_name = []
    for names in ITEMS:
        if names.endswith(".pth"):
            filename = os.path.join(FILE_DIR, names)
            file_name.append(filename)
            create_time.append(os.path.getctime(filename))  # 获取文件的修改时间
    # 将两个list转换为dict
    dit = dict(zip(create_time, file_name))
    # 根据dit的key对dit进行排序（变为list）
    dit = sorted(dit.items(), key=lambda d: d[-2], reverse=False)
    for i in range(image_num - max_image_saved):  # 删除文件个数
        file_location = dit[i][1]
        fileremove(file_location)
def move_model(FILE_DIR,max_image_saved = 5):
    ITEMS = os.listdir(FILE_DIR)  # FILE_DIR中的所有文件
    print('lent', len(ITEMS))
    model_num = len(ITEMS)
    if model_num > max_image_saved:
        check_model(ITEMS, model_num, FILE_DIR, max_image_saved)

'''计算文件夹容量 单位为MB'''
if __name__ == '__main__':
    FILE_DIR = '/home/jerry/shared/py_poject_wang/smok_detect/EfficientDet_torch12/logs/None'
    move_model(FILE_DIR,max_image_saved = 1)