From f4802a19c33499efb5842a7092e25c41dbb7d9d4 Mon Sep 17 00:00:00 2001 From: PaddlePaddle-Gardener Date: Fri, 7 Jan 2022 12:35:38 +0800 Subject: [PATCH] mirgate_1690 --- modules/video/multiple_object_tracking/fairmot_dla34/README.md | 2 +- .../video/multiple_object_tracking/fairmot_dla34/tracker.py | 3 ++- modules/video/multiple_object_tracking/jde_darknet53/README.md | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/modules/video/multiple_object_tracking/fairmot_dla34/README.md b/modules/video/multiple_object_tracking/fairmot_dla34/README.md index 22074a04..3ff8bd39 100644 --- a/modules/video/multiple_object_tracking/fairmot_dla34/README.md +++ b/modules/video/multiple_object_tracking/fairmot_dla34/README.md @@ -53,7 +53,7 @@ ``` - 通过命令行方式实现多目标追踪模型的调用，更多请见 [PaddleHub命令行指令](../../../../docs/docs_ch/tutorial/cmd_usage.rst) -- ### 2、代码示例 +- ### 2、预测代码示例 - ```python import paddlehub as hub diff --git a/modules/video/multiple_object_tracking/fairmot_dla34/tracker.py b/modules/video/multiple_object_tracking/fairmot_dla34/tracker.py index 5f91b122..f641527c 100644 --- a/modules/video/multiple_object_tracking/fairmot_dla34/tracker.py +++ b/modules/video/multiple_object_tracking/fairmot_dla34/tracker.py @@ -41,7 +41,8 @@ class StreamTracker(object): self.optimizer = None # build model - self.model = create(cfg.architecture) + with paddle.no_grad(): + self.model = create(cfg.architecture) self.status = {} self.start_epoch = 0 diff --git a/modules/video/multiple_object_tracking/jde_darknet53/README.md b/modules/video/multiple_object_tracking/jde_darknet53/README.md index 19746962..87e0755c 100644 --- a/modules/video/multiple_object_tracking/jde_darknet53/README.md +++ b/modules/video/multiple_object_tracking/jde_darknet53/README.md @@ -55,7 +55,7 @@ ``` - 通过命令行方式实现多目标追踪模型的调用，更多请见 [PaddleHub命令行指令](../../../../docs/docs_ch/tutorial/cmd_usage.rst) -- ### 2、代码示例 +- ### 2、预测代码示例 - ```python import paddlehub as hub -- Gitee