Fix random_perspective param bug in segment (#9512)
* fix random_perspective param bug when mosaic=False Signed-off-by: FeiGeChuanShu <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.ci Signed-off-by: FeiGeChuanShu <774074168@qq.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
1 file changed
+8
-11
@@ -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:
|
||||||
|
|||||||
Reference in new issue
Block a user