ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-22 09:37:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

tiny

Browse Source
This commit is contained in:
nttstar
2017-11-17 14:56:06 +08:00
parent eb918ee269
commit 4e36f3f35e
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/data.py
View File

@@ -557,7 +557,7 @@ class FaceImageIter2(io.DataIter):
_data = mx.ndarray.flip(data=_data, axis=1)
if self.nd_mean is not None:
_data = _data.astype('float32')
_data -= self.nd_mean
_data -= self.nd_mean
_data *= 0.0078125
#_npdata = _data.asnumpy()
#if landmark is not None:

5
src/train_softmax.py
View File

@@ -303,7 +303,8 @@ def train_net(args):
data_shape = (args.image_channel,112,96)
mean = [127.5,127.5,127.5]
if args.network[0]=='m' and args.num_layers==27:
#if args.network[0]=='m' and args.num_layers==27:
if args.network[0]=='m':
mean = None
if args.use_val:
@@ -588,7 +589,7 @@ def train_net(args):
#lr_steps = [40000, 70000, 90000]
lr_steps = [30000, 50000, 70000, 90000]
if args.loss_type==1:
lr_steps = [70000, 100000]
lr_steps = [100000, 140000, 160000]
else:
lr_steps = [int(x) for x in args.lr_steps.split(',')]
print('lr_steps', lr_steps)