diff --git a/modules/video/multiple_object_tracking/fairmot_dla34/README.md b/modules/video/multiple_object_tracking/fairmot_dla34/README.md index 22074a04ffedc228290d7b4de13bb3d6e651bf25..3ff8bd396a3bc60764241701856dfa20feb59459 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 5f91b1227595cde3bb1ed7f29dbeea039da48830..f641527ce94c8014db1afc0c5418bf6a278c352e 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 19746962688e10d2eaf7a3503e2e53e03462fc7a..87e0755cce6c439e4f0ebb04c6ef5e3f740d433a 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