Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
小马大咖
yolov7-obb
提交
b3de457d
Y
yolov7-obb
项目概览
小马大咖
/
yolov7-obb
与 Fork 源项目一致
从无法访问的项目Fork
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Y
yolov7-obb
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
b3de457d
编写于
2月 10, 2023
作者:
_白鹭先生_
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
修复显示
上级
8b37ea48
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
31 addition
and
6 deletion
+31
-6
utils/utils_rbox.py
utils/utils_rbox.py
+29
-2
yolo.py
yolo.py
+2
-4
未找到文件。
utils/utils_rbox.py
浏览文件 @
b3de457d
...
...
@@ -2,7 +2,7 @@
Author: [egrt]
Date: 2023-01-30 19:00:28
LastEditors: [egrt]
LastEditTime: 2023-02-
07 17:15:5
6
LastEditTime: 2023-02-
10 14:16:0
6
Description: Oriented Bounding Boxes utils
'''
...
...
@@ -185,4 +185,31 @@ def get_best_begin_point(coordinates):
"""
coordinates
=
list
(
map
(
get_best_begin_point_single
,
coordinates
.
tolist
()))
coordinates
=
np
.
array
(
coordinates
)
return
coordinates
\ No newline at end of file
return
coordinates
def
correct_rboxes
(
rboxes
,
image_shape
):
"""将polys按比例进行缩放
Args:
coordinate (ndarray): shape(n, 8).
Returns:
reorder coordinate (ndarray): shape(n, 8).
"""
polys
=
rbox2poly
(
rboxes
)
nh
,
nw
=
image_shape
polys
[:,
[
0
,
2
,
4
,
6
]]
*=
nw
polys
[:,
[
1
,
3
,
5
,
7
]]
*=
nh
rboxes
=
poly2rbox
(
polys
)
correct_polys
=
[]
for
rbox
in
rboxes
:
xc
,
yc
,
h
,
w
,
ag
=
rbox
[:
5
]
wx
,
wy
=
-
w
/
2
*
math
.
sin
(
ag
),
w
/
2
*
math
.
cos
(
ag
)
hx
,
hy
=
h
/
2
*
math
.
cos
(
ag
),
h
/
2
*
math
.
sin
(
ag
)
point_x1
,
point_y1
=
xc
-
wx
-
hx
,
yc
-
wy
-
hy
point_x2
,
point_y2
=
xc
-
wx
+
hx
,
yc
-
wy
+
hy
point_x3
,
point_y3
=
xc
+
wx
+
hx
,
yc
+
wy
+
hy
point_x4
,
point_y4
=
xc
+
wx
-
hx
,
yc
+
wy
-
hy
poly
=
np
.
array
([
point_x1
,
point_y1
,
point_x2
,
point_y2
,
point_x3
,
point_y3
,
point_x4
,
point_y4
])
correct_polys
.
append
(
poly
)
correct_polys
=
np
.
array
(
correct_polys
)
return
correct_polys
yolo.py
浏览文件 @
b3de457d
...
...
@@ -11,7 +11,7 @@ from nets.yolo import YoloBody
from
utils.utils
import
(
cvtColor
,
get_anchors
,
get_classes
,
preprocess_input
,
resize_image
,
show_config
)
from
utils.utils_bbox
import
DecodeBox
from
utils.utils_rbox
import
rbox2poly
,
poly2hbb
from
utils.utils_rbox
import
*
'''
训练自己的数据集必看注释!
'''
...
...
@@ -157,9 +157,7 @@ class YOLO(object):
top_label
=
np
.
array
(
results
[
0
][:,
7
],
dtype
=
'int32'
)
top_conf
=
results
[
0
][:,
5
]
*
results
[
0
][:,
6
]
top_rboxes
=
results
[
0
][:,
:
5
]
top_polys
=
rbox2poly
(
top_rboxes
)
top_polys
[:,
0
::
2
]
*=
image_shape
[
1
]
top_polys
[:,
1
::
2
]
*=
image_shape
[
0
]
top_polys
=
correct_rboxes
(
top_rboxes
,
image_shape
)
#---------------------------------------------------------#
# 设置字体与边框厚度
#---------------------------------------------------------#
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录