apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 09:07:10 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Modify model tests

Browse Source
This commit is contained in:
jiangjiajun
2022-11-05 07:54:16 +00:00
parent f00212aa42
commit cf8f53e36d
7 changed files with 111 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/models/test_pptracking.py
View File

@@ -23,16 +23,16 @@ import runtime_config as rc
def test_pptracking():
model_url = "https://bj.bcebos.com/fastdeploy/tests/pptracking.tgz"
input_url = "https://bj.bcebos.com/paddlehub/fastdeploy/person.mp4"
fd.download_and_decompress(model_url, ".")
fd.download(input_url, ".")
model_path = "pptracking/fairmot_hrnetv2_w18_dlafpn_30e_576x320"
fd.download_and_decompress(model_url, "resources")
fd.download(input_url, "resources")
model_path = "resources/pptracking/fairmot_hrnetv2_w18_dlafpn_30e_576x320"
# use default backend
runtime_option = fd.RuntimeOption()
model_file = os.path.join(model_path, "model.pdmodel")
params_file = os.path.join(model_path, "model.pdiparams")
config_file = os.path.join(model_path, "infer_cfg.yml")
model = fd.vision.tracking.PPTracking(model_file, params_file, config_file, runtime_option=rc.test_option)
cap = cv2.VideoCapture("./person.mp4")
cap = cv2.VideoCapture("./resources/person.mp4")
frame_id = 0
while True:
_, frame = cap.read()
@@ -40,7 +40,7 @@ def test_pptracking():
break
result = model.predict(frame)
# compare diff
expect = pickle.load(open("pptracking/frame" + str(frame_id) + ".pkl", "rb"))
expect = pickle.load(open("resources/pptracking/frame" + str(frame_id) + ".pkl", "rb"))
diff_boxes = np.fabs(np.array(expect["boxes"]) - np.array(result.boxes))
diff_scores = np.fabs(np.array(expect["scores"]) - np.array(result.scores))
diff = max(diff_boxes.max(), diff_scores.max())