Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
鹤冲天Pro
yolov7-obb
提交
e4c866e3
Y
yolov7-obb
项目概览
鹤冲天Pro
/
yolov7-obb
与 Fork 源项目一致
从无法访问的项目Fork
通知
1
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,体验更适合开发者的 AI 搜索 >>
提交
e4c866e3
编写于
2月 04, 2023
作者:
_白鹭先生_
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
修复马赛克数据增强
上级
72f1c6ff
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
17 addition
and
17 deletion
+17
-17
utils/dataloader.py
utils/dataloader.py
+17
-17
未找到文件。
utils/dataloader.py
浏览文件 @
e4c866e3
...
...
@@ -41,18 +41,18 @@ class YoloDataset(Dataset):
# 训练时进行数据的随机增强
# 验证时不进行数据的随机增强
#---------------------------------------------------#
#
if self.mosaic and self.rand() < self.mosaic_prob and self.epoch_now < self.epoch_length * self.special_aug_ratio:
lines
=
sample
(
self
.
annotation_lines
,
3
)
lines
.
append
(
self
.
annotation_lines
[
index
])
shuffle
(
lines
)
image
,
rbox
=
self
.
get_random_data_with_Mosaic
(
lines
,
self
.
input_shape
)
#
if self.mixup and self.rand() < self.mixup_prob:
#
lines = sample(self.annotation_lines, 1)
#
image_2, rbox_2 = self.get_random_data(lines[0], self.input_shape, random = self.train)
#
image, rbox = self.get_random_data_with_MixUp(image, rbox, image_2, rbox_2)
#
else:
#
image, rbox = self.get_random_data(self.annotation_lines[index], self.input_shape, random = self.train)
if
self
.
mosaic
and
self
.
rand
()
<
self
.
mosaic_prob
and
self
.
epoch_now
<
self
.
epoch_length
*
self
.
special_aug_ratio
:
lines
=
sample
(
self
.
annotation_lines
,
3
)
lines
.
append
(
self
.
annotation_lines
[
index
])
shuffle
(
lines
)
image
,
rbox
=
self
.
get_random_data_with_Mosaic
(
lines
,
self
.
input_shape
)
if
self
.
mixup
and
self
.
rand
()
<
self
.
mixup_prob
:
lines
=
sample
(
self
.
annotation_lines
,
1
)
image_2
,
rbox_2
=
self
.
get_random_data
(
lines
[
0
],
self
.
input_shape
,
random
=
self
.
train
)
image
,
rbox
=
self
.
get_random_data_with_MixUp
(
image
,
rbox
,
image_2
,
rbox_2
)
else
:
image
,
rbox
=
self
.
get_random_data
(
self
.
annotation_lines
[
index
],
self
.
input_shape
,
random
=
self
.
train
)
image
=
np
.
transpose
(
preprocess_input
(
np
.
array
(
image
,
dtype
=
np
.
float32
)),
(
2
,
0
,
1
))
rbox
=
np
.
array
(
rbox
,
dtype
=
np
.
float32
)
...
...
@@ -186,11 +186,11 @@ class YoloDataset(Dataset):
rbox
[:,
0
]
=
1
-
rbox
[:,
0
]
rbox
[:,
4
]
*=
-
1
# 查看旋转框是否正确
draw
=
ImageDraw
.
Draw
(
image
)
polys
=
rbox2poly
(
rbox
[...,
:
5
])
*
w
for
poly
in
polys
:
draw
.
polygon
(
xy
=
list
(
poly
))
image
.
show
()
#
draw = ImageDraw.Draw(image)
#
polys = rbox2poly(rbox[..., :5])*w
#
for poly in polys:
#
draw.polygon(xy=list(poly))
#
image.show()
return
image_data
,
rbox
def
merge_rboxes
(
self
,
rboxes
,
cutx
,
cuty
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录