Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
3adb1063
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看板
未验证
提交
3adb1063
编写于
9月 23, 2021
作者:
F
Feng Ni
提交者:
GitHub
9月 23, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix kitti metric deploy (#4227)
上级
93934304
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
5 addition
and
2 deletion
+5
-2
ppdet/data/source/category.py
ppdet/data/source/category.py
+5
-2
未找到文件。
ppdet/data/source/category.py
浏览文件 @
3adb1063
...
...
@@ -90,16 +90,19 @@ def get_categories(metric_type, anno_file=None, arch=None):
elif
metric_type
.
lower
()
in
[
'mot'
,
'motdet'
,
'reid'
]:
return
_mot_category
()
elif
metric_type
.
lower
()
in
[
'kitti'
,
'bdd100k'
]:
return
_mot_category
(
category
=
'car'
)
else
:
raise
ValueError
(
"unknown metric type {}"
.
format
(
metric_type
))
def
_mot_category
():
def
_mot_category
(
category
=
'person'
):
"""
Get class id to category id map and category id
to category name map of mot dataset
"""
label_map
=
{
'person'
:
0
}
label_map
=
{
category
:
0
}
label_map
=
sorted
(
label_map
.
items
(),
key
=
lambda
x
:
x
[
1
])
cats
=
[
l
[
0
]
for
l
in
label_map
]
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录