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

Rename img_size to imgsz (#86)

Browse source
This commit is contained in:
Glenn Jocher 2022-12-24 00:39:09 +01:00 • committed by GitHub
parent ae2443c210
commit 6432afc5f9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
25 changed files with 98 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

10
ultralytics/yolo/data/dataset.py
View file

@ -24,7 +24,7 @@ class YOLODataset(BaseDataset):
def __init__(
self,
img_path,
img_size=640,
imgsz=640,
label_path=None,
cache=False,
augment=True,
@ -41,7 +41,7 @@ class YOLODataset(BaseDataset):
self.use_segments = use_segments
self.use_keypoints = use_keypoints
assert not (self.use_segments and self.use_keypoints), "Can not use both segments and keypoints."
super().__init__(img_path, img_size, label_path, cache, augment, hyp, prefix, rect, batch_size, stride, pad,
super().__init__(img_path, imgsz, label_path, cache, augment, hyp, prefix, rect, batch_size, stride, pad,
single_cls)
def cache_labels(self, path=Path("./labels.cache")):
@ -128,11 +128,11 @@ class YOLODataset(BaseDataset):
# mosaic = False
if self.augment:
if mosaic:
transforms = mosaic_transforms(self.img_size, hyp)
transforms = mosaic_transforms(self.imgsz, hyp)
else:
transforms = affine_transforms(self.img_size, hyp)
transforms = affine_transforms(self.imgsz, hyp)
else:
transforms = Compose([LetterBox(new_shape=(self.img_size, self.img_size))])
transforms = Compose([LetterBox(new_shape=(self.imgsz, self.imgsz))])
transforms.append(
Format(bbox_format="xywh",
normalize=True,