Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
weixin_45480109
yolov3
提交
8d236eea
Y
yolov3
项目概览
weixin_45480109
/
yolov3
与 Fork 源项目一致
从无法访问的项目Fork
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Y
yolov3
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
8d236eea
编写于
12月 08, 2020
作者:
G
Glenn Jocher
提交者:
GitHub
12月 08, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Hybrid auto-labelling support (#1599)
* Introduce hybrid auto-labelling support * cleanup
上级
7e846c7d
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
6 addition
and
4 deletion
+6
-4
test.py
test.py
+6
-4
未找到文件。
test.py
浏览文件 @
8d236eea
...
...
@@ -33,7 +33,8 @@ def test(data,
dataloader
=
None
,
save_dir
=
Path
(
''
),
# for saving images
save_txt
=
False
,
# for auto-labelling
save_conf
=
False
,
save_hybrid
=
False
,
# for hybrid auto-labelling
save_conf
=
False
,
# save auto-label confidences
plots
=
True
,
log_imgs
=
0
):
# number of logged images
...
...
@@ -45,7 +46,6 @@ def test(data,
else
:
# called directly
set_logging
()
device
=
select_device
(
opt
.
device
,
batch_size
=
batch_size
)
save_txt
=
opt
.
save_txt
# save *.txt labels
# Directories
save_dir
=
Path
(
increment_path
(
Path
(
opt
.
project
)
/
opt
.
name
,
exist_ok
=
opt
.
exist_ok
))
# increment run
...
...
@@ -115,7 +115,7 @@ def test(data,
# Run NMS
targets
[:,
2
:]
*=
torch
.
Tensor
([
width
,
height
,
width
,
height
]).
to
(
device
)
# to pixels
lb
=
[
targets
[
targets
[:,
0
]
==
i
,
1
:]
for
i
in
range
(
nb
)]
if
save_
txt
else
[]
# for autolabelling
lb
=
[
targets
[
targets
[:,
0
]
==
i
,
1
:]
for
i
in
range
(
nb
)]
if
save_
hybrid
else
[]
# for autolabelling
t
=
time_synchronized
()
output
=
non_max_suppression
(
inf_out
,
conf_thres
=
conf_thres
,
iou_thres
=
iou_thres
,
labels
=
lb
)
t1
+=
time_synchronized
()
-
t
...
...
@@ -292,6 +292,7 @@ if __name__ == '__main__':
parser
.
add_argument
(
'--augment'
,
action
=
'store_true'
,
help
=
'augmented inference'
)
parser
.
add_argument
(
'--verbose'
,
action
=
'store_true'
,
help
=
'report mAP by class'
)
parser
.
add_argument
(
'--save-txt'
,
action
=
'store_true'
,
help
=
'save results to *.txt'
)
parser
.
add_argument
(
'--save-hybrid'
,
action
=
'store_true'
,
help
=
'save label+prediction hybrid results to *.txt'
)
parser
.
add_argument
(
'--save-conf'
,
action
=
'store_true'
,
help
=
'save confidences in --save-txt labels'
)
parser
.
add_argument
(
'--save-json'
,
action
=
'store_true'
,
help
=
'save a cocoapi-compatible JSON results file'
)
parser
.
add_argument
(
'--project'
,
default
=
'runs/test'
,
help
=
'save to project/name'
)
...
...
@@ -313,7 +314,8 @@ if __name__ == '__main__':
opt
.
single_cls
,
opt
.
augment
,
opt
.
verbose
,
save_txt
=
opt
.
save_txt
,
save_txt
=
opt
.
save_txt
|
opt
.
save_hybrid
,
save_hybrid
=
opt
.
save_hybrid
,
save_conf
=
opt
.
save_conf
,
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录