| @@ -6,7 +6,7 @@ from torch.autograd.variable import Variable | |||||
| from dface.core.models import PNet,RNet,ONet | from dface.core.models import PNet,RNet,ONet | ||||
| import dface.core.utils as utils | import dface.core.utils as utils | ||||
| import dface.core.image_tools as image_tools | import dface.core.image_tools as image_tools | ||||
| from __future__ import print_function | |||||
| def create_mtcnn_net(p_model_path=None, r_model_path=None, o_model_path=None, use_cuda=True): | def create_mtcnn_net(p_model_path=None, r_model_path=None, o_model_path=None, use_cuda=True): | ||||
| @@ -1,5 +1,6 @@ | |||||
| import os | import os | ||||
| import numpy as np | import numpy as np | ||||
| from __future__ import print_function | |||||
| class ImageDB(object): | class ImageDB(object): | ||||
| def __init__(self, image_annotation_file, prefix_path='', mode='train'): | def __init__(self, image_annotation_file, prefix_path='', mode='train'): | ||||
| @@ -1,7 +1,7 @@ | |||||
| import os | import os | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.prepare_data.assemble as assemble | import dface.prepare_data.assemble as assemble | ||||
| from __future__ import print_function | |||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| @@ -1,7 +1,7 @@ | |||||
| import os | import os | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.prepare_data.assemble as assemble | import dface.prepare_data.assemble as assemble | ||||
| from __future__ import print_function | |||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| @@ -1,7 +1,7 @@ | |||||
| import os | import os | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.prepare_data.assemble as assemble | import dface.prepare_data.assemble as assemble | ||||
| from __future__ import print_function | |||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| @@ -1,5 +1,4 @@ | |||||
| import argparse | import argparse | ||||
| import cv2 | import cv2 | ||||
| import numpy as np | import numpy as np | ||||
| from dface.core.detect import MtcnnDetector,create_mtcnn_net | from dface.core.detect import MtcnnDetector,create_mtcnn_net | ||||
| @@ -11,6 +10,8 @@ import cPickle | |||||
| from dface.core.utils import convert_to_square,IoU | from dface.core.utils import convert_to_square,IoU | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.core.vision as vision | import dface.core.vision as vision | ||||
| from __future__ import print_function | |||||
| def gen_onet_data(data_dir, anno_file, pnet_model_file, rnet_model_file, prefix_path='', use_cuda=True, vis=False): | def gen_onet_data(data_dir, anno_file, pnet_model_file, rnet_model_file, prefix_path='', use_cuda=True, vis=False): | ||||
| @@ -5,6 +5,7 @@ import os | |||||
| import numpy.random as npr | import numpy.random as npr | ||||
| from dface.core.utils import IoU | from dface.core.utils import IoU | ||||
| import dface.config as config | import dface.config as config | ||||
| from __future__ import print_function | |||||
| def gen_pnet_data(data_dir,anno_file,prefix): | def gen_pnet_data(data_dir,anno_file,prefix): | ||||
| @@ -1,6 +1,4 @@ | |||||
| import argparse | import argparse | ||||
| import cv2 | import cv2 | ||||
| import numpy as np | import numpy as np | ||||
| from dface.core.detect import MtcnnDetector,create_mtcnn_net | from dface.core.detect import MtcnnDetector,create_mtcnn_net | ||||
| @@ -12,6 +10,7 @@ import cPickle | |||||
| from dface.core.utils import convert_to_square,IoU | from dface.core.utils import convert_to_square,IoU | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.core.vision as vision | import dface.core.vision as vision | ||||
| from __future__ import print_function | |||||
| def gen_rnet_data(data_dir, anno_file, pnet_model_file, prefix_path='', use_cuda=True, vis=False): | def gen_rnet_data(data_dir, anno_file, pnet_model_file, prefix_path='', use_cuda=True, vis=False): | ||||
| @@ -7,7 +7,7 @@ import numpy.random as npr | |||||
| import argparse | import argparse | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.core.utils as utils | import dface.core.utils as utils | ||||
| from __future__ import print_function | |||||
| def gen_data(anno_file, data_dir, prefix): | def gen_data(anno_file, data_dir, prefix): | ||||
| @@ -8,7 +8,7 @@ import numpy.random as npr | |||||
| import argparse | import argparse | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.core.utils as utils | import dface.core.utils as utils | ||||
| from __future__ import print_function | |||||
| def gen_data(anno_file, data_dir, prefix): | def gen_data(anno_file, data_dir, prefix): | ||||
| @@ -8,7 +8,7 @@ import numpy.random as npr | |||||
| import argparse | import argparse | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.core.utils as utils | import dface.core.utils as utils | ||||
| from __future__ import print_function | |||||
| def gen_data(anno_file, data_dir, prefix): | def gen_data(anno_file, data_dir, prefix): | ||||
| @@ -11,6 +11,8 @@ import cPickle | |||||
| from dface.core.utils import convert_to_square,IoU | from dface.core.utils import convert_to_square,IoU | ||||
| import dface.config as config | import dface.config as config | ||||
| import dface.core.vision as vision | import dface.core.vision as vision | ||||
| from __future__ import print_function | |||||
| def gen_landmark48_data(data_dir, anno_file, pnet_model_file, rnet_model_file, prefix_path='', use_cuda=True, vis=False): | def gen_landmark48_data(data_dir, anno_file, pnet_model_file, rnet_model_file, prefix_path='', use_cuda=True, vis=False): | ||||
| @@ -5,9 +5,7 @@ from dface.core.models import PNet,RNet,ONet,LossFn | |||||
| import torch | import torch | ||||
| from torch.autograd import Variable | from torch.autograd import Variable | ||||
| import dface.core.image_tools as image_tools | import dface.core.image_tools as image_tools | ||||
| from __future__ import print_function | |||||
| def compute_accuracy(prob_cls, gt_cls): | def compute_accuracy(prob_cls, gt_cls): | ||||
| @@ -4,7 +4,7 @@ from dface.core.imagedb import ImageDB | |||||
| import dface.train_net.train as train | import dface.train_net.train as train | ||||
| import dface.config as config | import dface.config as config | ||||
| import os | import os | ||||
| from __future__ import print_function | |||||
| def train_net(annotation_file, model_store_path, | def train_net(annotation_file, model_store_path, | ||||
| @@ -4,7 +4,7 @@ from dface.core.imagedb import ImageDB | |||||
| from dface.train_net.train import train_pnet | from dface.train_net.train import train_pnet | ||||
| import dface.config as config | import dface.config as config | ||||
| import os | import os | ||||
| from __future__ import print_function | |||||
| def train_net(annotation_file, model_store_path, | def train_net(annotation_file, model_store_path, | ||||
| @@ -4,7 +4,7 @@ from dface.core.imagedb import ImageDB | |||||
| import dface.train_net.train as train | import dface.train_net.train as train | ||||
| import dface.config as config | import dface.config as config | ||||
| import os | import os | ||||
| from __future__ import print_function | |||||
| def train_net(annotation_file, model_store_path, | def train_net(annotation_file, model_store_path, | ||||