1kB-FaceImage/main.py at main · dasec/1kB-FaceImage · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import paths
import OFIQzmq as ofiq
from helpers import manipulate, create_dataset as cd
import CVLface.cvlface as cvlface

cd.create_cropped_dataset(paths.original_dataset, f"{paths.datasets_dir}/color/112/default/")
cd.create_grayscale_dataset(f"{paths.datasets_dir}/color/112/default/", f"{paths.datasets_dir}/gray/112/default/")

ofiq_instance = ofiq.init_ofiq_zmq()

manipulate.apply_manipulation_on_dataset(f"{paths.datasets_dir}/color/112/default/",
                                         f"{paths.datasets_dir}/color/112/ofiq_landmarks/",
                                         "ofiq_landmarks",
                                         ofiq_instance)

# This will download the necessary files for using CVLface for feature extraction
adaface_model = cvlface.CvlfaceFrAdapter()
adaface_model.load_fr_feature_extractor_model()