Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
22736876
P
PaddleDetection
项目概览
s920243400
/
PaddleDetection
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleDetection
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
22736876
编写于
3月 22, 2022
作者:
W
wangguanzhong
提交者:
GitHub
3月 22, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refine pptracking import (#5421)
上级
8622966f
变更
7
隐藏空白更改
内联
并排
Showing
7 changed file
with
14 addition
and
15 deletion
+14
-15
deploy/pphuman/pipeline.py
deploy/pphuman/pipeline.py
+1
-1
deploy/pptracking/python/det_infer.py
deploy/pptracking/python/det_infer.py
+3
-3
deploy/pptracking/python/mot/visualize.py
deploy/pptracking/python/mot/visualize.py
+0
-0
deploy/pptracking/python/mot_jde_infer.py
deploy/pptracking/python/mot_jde_infer.py
+4
-4
deploy/pptracking/python/mot_sde_infer.py
deploy/pptracking/python/mot_sde_infer.py
+4
-4
deploy/pptracking/python/mot_utils.py
deploy/pptracking/python/mot_utils.py
+1
-2
deploy/python/mot_sde_infer.py
deploy/python/mot_sde_infer.py
+1
-1
未找到文件。
deploy/pphuman/pipeline.py
浏览文件 @
22736876
...
@@ -40,7 +40,7 @@ from python.preprocess import decode_image
...
@@ -40,7 +40,7 @@ from python.preprocess import decode_image
from
python.visualize
import
visualize_box_mask
,
visualize_attr
,
visualize_pose
,
visualize_action
from
python.visualize
import
visualize_box_mask
,
visualize_attr
,
visualize_pose
,
visualize_action
from
pptracking.python.mot_sde_infer
import
SDE_Detector
from
pptracking.python.mot_sde_infer
import
SDE_Detector
from
pptracking.python.visualize
import
plot_tracking
from
pptracking.python.
mot.
visualize
import
plot_tracking
class
Pipeline
(
object
):
class
Pipeline
(
object
):
...
...
deploy/pptracking/python/det_infer.py
浏览文件 @
22736876
...
@@ -33,8 +33,8 @@ sys.path.insert(0, parent_path)
...
@@ -33,8 +33,8 @@ sys.path.insert(0, parent_path)
from
benchmark_utils
import
PaddleInferBenchmark
from
benchmark_utils
import
PaddleInferBenchmark
from
picodet_postprocess
import
PicoDetPostProcess
from
picodet_postprocess
import
PicoDetPostProcess
from
preprocess
import
preprocess
,
Resize
,
NormalizeImage
,
Permute
,
PadStride
,
LetterBoxResize
,
decode_image
from
preprocess
import
preprocess
,
Resize
,
NormalizeImage
,
Permute
,
PadStride
,
LetterBoxResize
,
decode_image
from
visualize
import
visualize_box_mask
from
mot.
visualize
import
visualize_box_mask
from
utils
import
argsparser
,
Timer
,
get_current_memory_mb
from
mot_
utils
import
argsparser
,
Timer
,
get_current_memory_mb
# Global dictionary
# Global dictionary
SUPPORT_MODELS
=
{
SUPPORT_MODELS
=
{
...
@@ -269,7 +269,7 @@ class Detector(object):
...
@@ -269,7 +269,7 @@ class Detector(object):
if
not
os
.
path
.
exists
(
self
.
output_dir
):
if
not
os
.
path
.
exists
(
self
.
output_dir
):
os
.
makedirs
(
self
.
output_dir
)
os
.
makedirs
(
self
.
output_dir
)
out_path
=
os
.
path
.
join
(
self
.
output_dir
,
video_out_name
)
out_path
=
os
.
path
.
join
(
self
.
output_dir
,
video_out_name
)
fourcc
=
cv2
.
VideoWriter_fourcc
(
*
'mp4v'
)
fourcc
=
cv2
.
VideoWriter_fourcc
(
*
'mp4v'
)
writer
=
cv2
.
VideoWriter
(
out_path
,
fourcc
,
fps
,
(
width
,
height
))
writer
=
cv2
.
VideoWriter
(
out_path
,
fourcc
,
fps
,
(
width
,
height
))
index
=
1
index
=
1
while
(
1
):
while
(
1
):
...
...
deploy/pptracking/python/visualize.py
→
deploy/pptracking/python/
mot/
visualize.py
浏览文件 @
22736876
文件已移动
deploy/pptracking/python/mot_jde_infer.py
浏览文件 @
22736876
...
@@ -22,7 +22,7 @@ import paddle
...
@@ -22,7 +22,7 @@ import paddle
from
benchmark_utils
import
PaddleInferBenchmark
from
benchmark_utils
import
PaddleInferBenchmark
from
preprocess
import
decode_image
from
preprocess
import
decode_image
from
utils
import
argsparser
,
Timer
,
get_current_memory_mb
from
mot_
utils
import
argsparser
,
Timer
,
get_current_memory_mb
from
det_infer
import
Detector
,
get_test_images
,
print_arguments
,
bench_log
,
PredictConfig
from
det_infer
import
Detector
,
get_test_images
,
print_arguments
,
bench_log
,
PredictConfig
# add python path
# add python path
...
@@ -31,8 +31,8 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 2)))
...
@@ -31,8 +31,8 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 2)))
sys
.
path
.
insert
(
0
,
parent_path
)
sys
.
path
.
insert
(
0
,
parent_path
)
from
mot
import
JDETracker
from
mot
import
JDETracker
from
utils
import
MOTTimer
,
write_mot_results
from
mot.
utils
import
MOTTimer
,
write_mot_results
from
visualize
import
plot_tracking
,
plot_tracking_dict
from
mot.
visualize
import
plot_tracking
,
plot_tracking_dict
# Global dictionary
# Global dictionary
MOT_JDE_SUPPORT_MODELS
=
{
MOT_JDE_SUPPORT_MODELS
=
{
...
@@ -264,7 +264,7 @@ class JDE_Detector(Detector):
...
@@ -264,7 +264,7 @@ class JDE_Detector(Detector):
if
not
os
.
path
.
exists
(
self
.
output_dir
):
if
not
os
.
path
.
exists
(
self
.
output_dir
):
os
.
makedirs
(
self
.
output_dir
)
os
.
makedirs
(
self
.
output_dir
)
out_path
=
os
.
path
.
join
(
self
.
output_dir
,
video_out_name
)
out_path
=
os
.
path
.
join
(
self
.
output_dir
,
video_out_name
)
fourcc
=
cv2
.
VideoWriter_fourcc
(
*
'mp4v'
)
fourcc
=
cv2
.
VideoWriter_fourcc
(
*
'mp4v'
)
writer
=
cv2
.
VideoWriter
(
out_path
,
fourcc
,
fps
,
(
width
,
height
))
writer
=
cv2
.
VideoWriter
(
out_path
,
fourcc
,
fps
,
(
width
,
height
))
frame_id
=
1
frame_id
=
1
...
...
deploy/pptracking/python/mot_sde_infer.py
浏览文件 @
22736876
...
@@ -24,17 +24,17 @@ import paddle
...
@@ -24,17 +24,17 @@ import paddle
from
benchmark_utils
import
PaddleInferBenchmark
from
benchmark_utils
import
PaddleInferBenchmark
from
preprocess
import
decode_image
from
preprocess
import
decode_image
from
.utils
import
argsparser
,
Timer
,
get_current_memory_mb
,
_is_valid_video
,
video2frames
from
.det_infer
import
Detector
,
get_test_images
,
print_arguments
,
bench_log
,
PredictConfig
,
load_predictor
# add python path
# add python path
import
sys
import
sys
parent_path
=
os
.
path
.
abspath
(
os
.
path
.
join
(
__file__
,
*
([
'..'
]
*
2
)))
parent_path
=
os
.
path
.
abspath
(
os
.
path
.
join
(
__file__
,
*
([
'..'
])))
sys
.
path
.
insert
(
0
,
parent_path
)
sys
.
path
.
insert
(
0
,
parent_path
)
from
det_infer
import
Detector
,
get_test_images
,
print_arguments
,
bench_log
,
PredictConfig
,
load_predictor
from
mot_utils
import
argsparser
,
Timer
,
get_current_memory_mb
,
video2frames
,
_is_valid_video
from
mot.tracker
import
JDETracker
,
DeepSORTTracker
from
mot.tracker
import
JDETracker
,
DeepSORTTracker
from
mot.utils
import
MOTTimer
,
write_mot_results
,
flow_statistic
,
get_crops
,
clip_box
from
mot.utils
import
MOTTimer
,
write_mot_results
,
flow_statistic
,
get_crops
,
clip_box
from
.visualize
import
plot_tracking
,
plot_tracking_dict
from
mot
.visualize
import
plot_tracking
,
plot_tracking_dict
from
mot.mtmct.utils
import
parse_bias
from
mot.mtmct.utils
import
parse_bias
from
mot.mtmct.postprocess
import
trajectory_fusion
,
sub_cluster
,
gen_res
,
print_mtmct_result
from
mot.mtmct.postprocess
import
trajectory_fusion
,
sub_cluster
,
gen_res
,
print_mtmct_result
...
...
deploy/pptracking/python/utils.py
→
deploy/pptracking/python/
mot_
utils.py
浏览文件 @
22736876
...
@@ -147,7 +147,6 @@ def argsparser():
...
@@ -147,7 +147,6 @@ def argsparser():
return
parser
return
parser
class
Times
(
object
):
class
Times
(
object
):
def
__init__
(
self
):
def
__init__
(
self
):
self
.
time
=
0.
self
.
time
=
0.
...
@@ -241,7 +240,7 @@ class Timer(Times):
...
@@ -241,7 +240,7 @@ class Timer(Times):
total_time
=
pre_time
+
infer_time
+
post_time
total_time
=
pre_time
+
infer_time
+
post_time
if
self
.
with_tracker
:
if
self
.
with_tracker
:
dic
[
'tracking_time_s'
]
=
round
(
track_time
/
max
(
1
,
self
.
img_num
),
dic
[
'tracking_time_s'
]
=
round
(
track_time
/
max
(
1
,
self
.
img_num
),
4
)
if
average
else
track_time
4
)
if
average
else
track_time
total_time
=
total_time
+
track_time
total_time
=
total_time
+
track_time
dic
[
'total_time_s'
]
=
round
(
total_time
,
4
)
dic
[
'total_time_s'
]
=
round
(
total_time
,
4
)
return
dic
return
dic
...
...
deploy/python/mot_sde_infer.py
浏览文件 @
22736876
...
@@ -32,7 +32,7 @@ sys.path.insert(0, parent_path)
...
@@ -32,7 +32,7 @@ sys.path.insert(0, parent_path)
from
pptracking.python.mot
import
JDETracker
from
pptracking.python.mot
import
JDETracker
from
pptracking.python.mot.utils
import
MOTTimer
,
write_mot_results
from
pptracking.python.mot.utils
import
MOTTimer
,
write_mot_results
from
pptracking.python.visualize
import
plot_tracking
,
plot_tracking_dict
from
pptracking.python.
mot.
visualize
import
plot_tracking
,
plot_tracking_dict
class
SDE_Detector
(
Detector
):
class
SDE_Detector
(
Detector
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录