Beispiel #1
0
from pathlib import Path
from config import get_config
from data.data_pipe import load_bin, load_mx_rec
import argparse

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='for face verification')
    parser.add_argument("-r", "--rec_path", help="mxnet record file path",default='ms1m-retinaface-t1', type=str)
    args = parser.parse_args()
    conf = get_config()
    rec_path = conf.data_path/args.rec_path
    load_mx_rec(rec_path)
    
    bin_files = ['agedb_30', 'cfp_fp', 'lfw', 'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp']
    
    for i in range(len(bin_files)):
        load_bin(rec_path/(bin_files[i]+'.bin'), rec_path/bin_files[i], conf.test_transform)
from config import get_config
from data.data_pipe import load_bin, load_mx_rec
import argparse

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description = 'for extracting faces_emore data')
    parser.add_argument("-r", "--rec_path", help="mxnet record file path",default = 'faces_emore', type = str)
    args = parser.parse_args()
    conf = get_config()
    rec_path = conf.data_path/args.rec_path
    load_mx_rec(rec_path)
    
    bin_files = ['agedb_30', 'cfp_fp', 'lfw', 'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp']
    
    for i in range(len(bin_files)):
        load_bin(rec_path/(bin_files[i] + '.bin'), rec_path/bin_files[i], conf.test_transform)
Beispiel #3
0
from pathlib import Path
from config import get_config
from data.data_pipe import load_bin, load_mx_rec_custom
import argparse

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='for face verification')
    parser.add_argument(
        "-r", "--rec_path", help="mxnet record file path", default='facescrub2', type=str)
    args = parser.parse_args()
    conf = get_config()
    rec_path = Path("/media/2tb/data/DeepGlint/faces_glint")
    load_mx_rec_custom(rec_path)

    # bin_files = ['agedb_30', 'cfp_fp', 'lfw',
    #              'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp']

    # for i in range(len(bin_files)):
    #     load_bin(rec_path/(bin_files[i]+'.bin'),
    #              rec_path/bin_files[i], conf.test_transform)

    load_bin(rec_path/("lfw"+'.bin'),
                 conf.emore_folder/"lfw", conf.test_transform)
Beispiel #4
0
from data.data_pipe import load_bin, load_mx_rec
from pathlib import Path
from config import configurations
import argparse
from torchvision import transforms as trans
import os

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description = 'for extracting faces_emore data')
    parser.add_argument("-r", "--rec_path", help="mxnet record file path",default = 'faces_emore', type = str)
    args = parser.parse_args()
    rec_path = Path(configurations[1]['DATA_ROOT'])
    load_mx_rec(rec_path)
    
    bin_files = ['agedb_30', 'cfp_fp', 'lfw', 'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp']
    
    for i in range(len(bin_files)):
        load_bin(rec_path/(bin_files[i] + '.bin'), rec_path/bin_files[i], trans.Compose([
            trans.ToTensor(),
            trans.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])
        ]))
Beispiel #5
0
import os
from pathlib import Path
from data.data_pipe import load_bin, load_mx_rec

if __name__ == '__main__':
   
    rec_path = 'data/faces_emore'
    load_mx_rec(rec_path)
    bin_files = ['agedb_30', 'cfp_fp', 'lfw']
    for i in range(len(bin_files)):
        load_bin(os.path.join(rec_path, bin_files[i]+'.bin'), os.path.join(rec_path, bin_files[i]))
from pathlib import Path
from config import get_config
from data.data_pipe import load_bin, load_mx_rec
import argparse
import args
import os
if __name__ == '__main__':

    conf = get_config()
    root_paths = conf.data_path / args.rec_path
    print('load mx rec from paths {}'.format(root_paths))
    load_mx_rec(root_paths)

    bin_files = [
        'agedb_30', 'cfp_fp', 'lfw', 'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp'
    ]
    print('load bin files from load bin \n')
    for bin_file in bin_files:
        full_path = root_paths / (bin_file + '.bin')
        folder_path = root_paths / bin_file
        print("full path: {} - status: {}".format(full_path.exists()))
        print("folder path: {} - status: {}".format(folder_path.exists()))
        load_bin(full_path, folder_path, conf.test_transform)
Beispiel #7
0
    parser.add_argument("-r",
                        "--rec_path",
                        help="mxnet record file path",
                        default='faces_emore',
                        type=str)
    args = parser.parse_args()
    conf = get_config()
    rec_path = conf.data_path / args.rec_path
    #    load_mx_rec(rec_path)

    #    bin_files = ['agedb_30', 'cfp_fp', 'lfw', 'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp']
    bin_files = ['cfp_fp', 'lfw', 'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp']

    #    for i in range(len(bin_files)):
    #        load_bin(rec_path/(bin_files[i]+'.bin'), rec_path/bin_files[i], conf.test_transform)
    #    load_bin('/home/zzg/Datasets/faces_emore/agedb_30.bin', '/home/zzg/Datasets/faces_emore/agedb_30', conf.test_transform)
    load_bin('/home/zzg/Datasets/faces_emore/cfp_fp.bin',
             '/home/zzg/Datasets/faces_emore/cfp_fp', conf.test_transform)
    load_bin('/home/zzg/Datasets/faces_emore/lfw.bin',
             '/home/zzg/Datasets/faces_emore/lfw', conf.test_transform)
    load_bin('/home/zzg/Datasets/faces_emore/lfw.bin',
             '/home/zzg/Datasets/faces_emore/lfw', conf.test_transform)
    load_bin('/home/zzg/Datasets/faces_emore/calfw.bin',
             '/home/zzg/Datasets/faces_emore/calfw', conf.test_transform)
    load_bin('/home/zzg/Datasets/faces_emore/cfp_ff.bin',
             '/home/zzg/Datasets/faces_emore/cfp_ff', conf.test_transform)
    load_bin('/home/zzg/Datasets/faces_emore/cplfw.bin',
             '/home/zzg/Datasets/faces_emore/cplfw', conf.test_transform)
    load_bin('/home/zzg/Datasets/faces_emore/vgg2_fp.bin',
             '/home/zzg/Datasets/faces_emore/vgg2_fp', conf.test_transform)
                        help="mxnet record file path",
                        default='faces_webface',
                        type=str)
    parser.add_argument(
        "-s",
        "--save_path",
        help="save path",
        default='/home/jason/Datasets/InsightFace/faces_webface_112x112',
        type=str)
    parser.add_argument("-i",
                        "--img_path",
                        help="img path",
                        default='/home/jason/Datasets/InsightFace',
                        type=str)

    args = parser.parse_args()
    conf = get_config()
    rec_path = conf.data_path / args.rec_path

    # load_mx_rec(rec_path)
    save_path = Path(args.save_path)
    load_mx_rec(rec_path, save_path, 0)

    # bin_files = ['agedb_30', 'cfp_fp', 'lfw', 'calfw', 'cfp_ff', 'cplfw', 'vgg2_fp']
    bin_files = ['lfw', 'calfw', 'cplfw']
    img_path = Path(args.img_path)
    for i in range(len(bin_files)):
        # load_bin(rec_path/(bin_files[i]+'.bin'), rec_path/bin_files[i], conf.test_transform)
        load_bin(rec_path / (bin_files[i] + '.bin'), rec_path / bin_files[i],
                 conf.test_transform, img_path / (bin_files[i] + '_112x112'))