Unverified 提交 f8b74631 authored 作者: FeiGeChuanShu's avatar FeiGeChuanShu 提交者: GitHub

Fix `random_perspective` param bug in segment (#9512)

* fix random_perspective param bug when mosaic=False Signed-off-by: 's avatarFeiGeChuanShu <774074168@qq.com> * Update dataloaders.py * Update dataloaders.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ciSigned-off-by: 's avatarFeiGeChuanShu <774074168@qq.com> Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
上级 095f601d
...@@ -140,17 +140,14 @@ class LoadImagesAndLabelsAndMasks(LoadImagesAndLabels): # for training/testing ...@@ -140,17 +140,14 @@ class LoadImagesAndLabelsAndMasks(LoadImagesAndLabels): # for training/testing
labels[:, 1:] = xywhn2xyxy(labels[:, 1:], ratio[0] * w, ratio[1] * h, padw=pad[0], padh=pad[1]) labels[:, 1:] = xywhn2xyxy(labels[:, 1:], ratio[0] * w, ratio[1] * h, padw=pad[0], padh=pad[1])
if self.augment: if self.augment:
img, labels, segments = random_perspective( img, labels, segments = random_perspective(img,
img, labels,
labels, segments=segments,
segments=segments, degrees=hyp["degrees"],
degrees=hyp["degrees"], translate=hyp["translate"],
translate=hyp["translate"], scale=hyp["scale"],
scale=hyp["scale"], shear=hyp["shear"],
shear=hyp["shear"], perspective=hyp["perspective"])
perspective=hyp["perspective"],
return_seg=True,
)
nl = len(labels) # number of labels nl = len(labels) # number of labels
if nl: if nl:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论