HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Remove unnecessary assignments (#15582)

Browse source
This commit is contained in:
Laughing 2024-08-15 03:32:37 +08:00 • committed by GitHub
parent 38bb58ae12
commit be9fe124e3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/data/augment.py
View file

@ -539,7 +539,6 @@ class Mosaic(BaseMixTransform):
assert 0 <= p <= 1.0, f"The probability should be in range [0, 1], but got {p}." assert 0 <= p <= 1.0, f"The probability should be in range [0, 1], but got {p}."
assert n in {4, 9}, "grid must be equal to 4 or 9." assert n in {4, 9}, "grid must be equal to 4 or 9."
super().__init__(dataset=dataset, p=p) super().__init__(dataset=dataset, p=p)
self.dataset = dataset
self.imgsz = imgsz self.imgsz = imgsz
self.border = (-imgsz // 2, -imgsz // 2) # width, height self.border = (-imgsz // 2, -imgsz // 2) # width, height
self.n = n self.n = n
@ -1692,7 +1691,6 @@ class CopyPaste:
instances.convert_bbox(format="xyxy") instances.convert_bbox(format="xyxy")
instances.denormalize(w, h) instances.denormalize(w, h)
if self.p and len(instances.segments): if self.p and len(instances.segments):
n = len(instances)
_, w, _ = im.shape # height, width, channels _, w, _ = im.shape # height, width, channels
im_new = np.zeros(im.shape, np.uint8) im_new = np.zeros(im.shape, np.uint8)