ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-19 07:11:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

read image_size from property file

Browse Source
This commit is contained in:
Jia Guo
2017-12-08 13:54:45 +08:00
parent 8b9e9c58d3
commit 6a0cd1aa1f
6 changed files with 40 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/eval/do_ver.sh Normal file → Executable file
View File

@@ -1,2 +1,3 @@
python -u verification.py --data-dir /opt/jiaguo/faces_vgg_112x112 --image-size 112,112 --model '../../model/softmax1010d3-r101-p0_0_96_112_0,21|22|32'
#python -u verification.py --gpu 0 --data-dir /opt/jiaguo/faces_vgg_112x112 --image-size 112,112 --model '../../model/softmax1010d3-r101-p0_0_96_112_0,21|22|32' --target agedb_30
python -u verification.py --gpu 0 --data-dir /opt/jiaguo/faces_normed --image-size 112,96 --model '../../model31/sphere-m51-p0_0_96_112_0,90' --target agedb_30 --batch-size 128

11
src/eval/verification.py
View File

@@ -38,6 +38,8 @@ import pickle
from sklearn.decomposition import PCA
import mxnet as mx
from mxnet import ndarray as nd
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'common'))
import face_image
@@ -250,13 +252,14 @@ if __name__ == '__main__':
# general
parser.add_argument('--data-dir', default='', help='')
parser.add_argument('--model', default='../model/softmax,50', help='path to load model.')
parser.add_argument('--image-size', default='112,96', help='image size.')
parser.add_argument('--target', default='lfw,cfp_ff,cfp_fp', help='test targets.')
parser.add_argument('--target', default='lfw,cfp_ff,cfp_fp,agedb_30', help='test targets.')
parser.add_argument('--gpu', default=0, type=int, help='gpu id')
parser.add_argument('--batch-size', default=128, type=int, help='')
parser.add_argument('--batch-size', default=32, type=int, help='')
args = parser.parse_args()
image_size = [int(x) for x in args.image_size.split(',')]
prop = face_image.load_property(args.data_dir)
image_size = prop.image_size
print('image_size', image_size)
ctx = mx.gpu(args.gpu)
nets = []
prefix = args.model.split(',')[0]