centerface interface change

4 years ago · 122d52200e
parent 84957cc4a7
commit 122d52200e
2 changed files with 4 additions and 1 deletions
--- a/model_zoo/official/cv/centerface/dependency/centernet/src/lib/external/Makefile
+++ b/model_zoo/official/cv/centerface/dependency/centernet/src/lib/external/Makefile
@ -0,0 +1,3 @@
+all:
+	python setup.py build_ext --inplace
+	rm -rf build
--- a/model_zoo/official/cv/centerface/scripts/test.sh
+++ b/model_zoo/official/cv/centerface/scripts/test.sh
@ -51,7 +51,7 @@ dataset_path=$dataset_root/centerface/images/val/images/
 ground_truth_mat=$dataset_root/centerface/ground_truth/val.mat
 save_path=$root/output/centerface/
 device_id=0
-ckpt="0.ckpt" # the model saved for epoch=125
+ckpt="0-125_24750.ckpt" # the model saved for epoch=125

 if [ $# == 1 ]
 then