diff --git a/model_zoo/official/cv/centerface/dependency/centernet/src/lib/external/Makefile b/model_zoo/official/cv/centerface/dependency/centernet/src/lib/external/Makefile new file mode 100644 index 00000000000..a48239850fb --- /dev/null +++ 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 diff --git a/model_zoo/official/cv/centerface/scripts/test.sh b/model_zoo/official/cv/centerface/scripts/test.sh index c3470c71f41..4d623bd608f 100644 --- 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