From 394578f039e461f2f7561977f6b5de32f28ece4a Mon Sep 17 00:00:00 2001 From: panjinquan Date: Wed, 22 Sep 2021 18:32:35 +0800 Subject: [PATCH] switch CPU --- src/main_for_pose.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/main_for_pose.cpp b/src/main_for_pose.cpp index 8a6dbdc..6d333be 100644 --- a/src/main_for_pose.cpp +++ b/src/main_for_pose.cpp @@ -22,7 +22,7 @@ void test_coco_person() { const int num_thread = 1; const float score_thresh = 0.5; - TNNDevice device_id = TNNGPU;//运行设备 + TNNDevice device_id = TNNCPU;//运行设备 ModelParam model_param = COCO_PERSON_PARAM;//模型参数 KeyPointDetector *detector = new KeyPointDetector(model_path, model_param,