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

ultralytics 8.0.103 minor fixes (#2634)

Browse source 
Co-authored-by: kssion <appplus@icloud.com>
Co-authored-by: Eli Spizzichino <diramazioni@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: xbkaishui <xxkaishui@gmail.com>
This commit is contained in:
Glenn Jocher 2023-05-16 01:39:52 +02:00 • committed by GitHub
parent e2e3e367a2
commit 7884098857
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 7 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/data/augment.py
View file

@ -128,7 +128,7 @@ class Mosaic(BaseMixTransform):
n (int, optional): The grid size, either 4 (for 2x2) or 9 (for 3x3).
"""
def __init__(self, dataset, imgsz=640, p=1.0, n=9):
def __init__(self, dataset, imgsz=640, p=1.0, n=4):
"""Initializes the object with a dataset, image size, probability, and border."""
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.'