def sharpness_clean(imagelist,save_path,dis=40): if not os.path.exists(save_path): os.mkdir(save_path) fr = open(imagelist,'r') lines = fr.readlines() num = 0; for line in lines: words = line.strip('\n').split('.') suff = words[1] new_filename = "%s%09d%s"%(save_path+'/',num,'.'+suff) d = sharpness.detection(line.strip('\n')) if d>dis: shutil.copyfile(line.strip('\n'),new_filename) num+=1 print u'清晰度清除已经完成,共保留%d张图片'%(num)
def sharpness_clean(imagelist, save_path, dis=40): if not os.path.exists(save_path): os.mkdir(save_path) fr = open(imagelist, 'r') lines = fr.readlines() num = 0 for line in lines: words = line.strip('\n').split('.') suff = words[1] new_filename = "%s%09d%s" % (save_path + '/', num, '.' + suff) d = sharpness.detection(line.strip('\n')) if d > dis: shutil.copyfile(line.strip('\n'), new_filename) num += 1 print u'清晰度清除已经完成,共保留%d张图片' % (num)
imagelist----保存图片路径的文件 out----保存特征文件的文件夹路径 ''' def ext_feature(proto,model,imagelist,out): if not os.path.exists(out): os.mkdir(out) command_line = "%s %s %s %s %s"%("feature.exe","--proto="+proto,\ "--model="+model,"--imagelist="+imagelist,"--out="+out) os.system(command_line) print u'特征提取完毕' if __name__ == "__main__": vedio_path = 'E:/1-18/12mm/2.avi' save_path = 'E:/practical_face/result/code/video/Output/Pictures' suff = '.bmp' a = sharpness.detection('E:/wu_test/pictures1/000000000.jpg') print a ''' pic_dath = 'E:/practical_face/result/code/video/Output/Pictures' imagelist = 'E:/practical_face/result/code/video/Output/imagelist.txt' save_path = 'E:/practical_face/result/code/video/Output/sharpness_clean' text_process.creat_imagelist_NoLabel(pic_dath,imagelist) sharpness_clean(imagelist,save_path,40) ''' proto = 'C:/Users/Administrator/Desktop/model/net_deploy.prototxt' model = 'C:/Users/Administrator/Desktop/model/net.caffemodel' imagelist = 'E:/practical_face/result/code/video/Output/imagelist1.txt' dir_path = 'E:/practical_face/result/code/video/Output/sharpness_clean' out = 'E:/practical_face/result/code/video/Output/txt' text_process.creat_imagelist_NoLabel(dir_path,imagelist)
def ext_feature(proto, model, imagelist, out): if not os.path.exists(out): os.mkdir(out) command_line = "%s %s %s %s %s"%("feature.exe","--proto="+proto,\ "--model="+model,"--imagelist="+imagelist,"--out="+out) os.system(command_line) print u'特征提取完毕' if __name__ == "__main__": vedio_path = 'E:/1-18/12mm/2.avi' save_path = 'E:/practical_face/result/code/video/Output/Pictures' suff = '.bmp' a = sharpness.detection('E:/wu_test/pictures1/000000000.jpg') print a ''' pic_dath = 'E:/practical_face/result/code/video/Output/Pictures' imagelist = 'E:/practical_face/result/code/video/Output/imagelist.txt' save_path = 'E:/practical_face/result/code/video/Output/sharpness_clean' text_process.creat_imagelist_NoLabel(pic_dath,imagelist) sharpness_clean(imagelist,save_path,40) ''' proto = 'C:/Users/Administrator/Desktop/model/net_deploy.prototxt' model = 'C:/Users/Administrator/Desktop/model/net.caffemodel' imagelist = 'E:/practical_face/result/code/video/Output/imagelist1.txt' dir_path = 'E:/practical_face/result/code/video/Output/sharpness_clean' out = 'E:/practical_face/result/code/video/Output/txt' text_process.creat_imagelist_NoLabel(dir_path, imagelist) ext_feature(proto, model, imagelist, out)