https://github.com/yule-li/CosFace
Raw File
Tip revision: 4a1857ec586fe9ddb6175165330b1ae7a9b9a79a authored by yule-li on 26 September 2018, 13:18:06 UTC
add dataset-112x112 in readme
Tip revision: 4a1857e
test.sh
V="v1"
if test ${V} = "v1"
then
  MODEL_DIR=models/model-20180309-083949.ckpt-60000
  TEST_DATA=dataset/lfw-112X96
  EMBEDDING_SIZE=512
  FC_BN=''
  PREWHITEN='--prewhiten'
  IMAGE_WIDTH=96
else
  MODEL_DIR=models/model-20180626-205832.ckpt-60000
  TEST_DATA=dataset/lfw-112x112
  EMBEDDING_SIZE=1024  
  FC_BN='--fc_bn'
  PREWHITEN=''
  IMAGE_WIDTH=112
fi
IMAGE_HEIGHT=112
CUDA_VISIBLE_DEVICES=1 python test/test.py ${TEST_DATA} ${MODEL_DIR} --lfw_file_ext jpg --network_type sphere_network --embedding_size ${EMBEDDING_SIZE} ${FC_BN} ${PREWHITEN} --image_height ${IMAGE_HEIGHT} --image_width ${IMAGE_WIDTH}
back to top