From 122d52200e6909014360d6d1df3217390e9e70a1 Mon Sep 17 00:00:00 2001 From: linqingke Date: Wed, 18 Nov 2020 16:54:18 +0800 Subject: [PATCH] centerface interface change --- .../centerface/dependency/centernet/src/lib/external/Makefile | 3 +++ model_zoo/official/cv/centerface/scripts/test.sh | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) create mode 100644 model_zoo/official/cv/centerface/dependency/centernet/src/lib/external/Makefile 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