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

Clean up unused imgsz (#7771)

Browse source
This commit is contained in:
Laughing 2024-01-24 01:50:01 +08:00 • committed by GitHub
parent f56dd0f48e
commit 67ae86f006
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 10 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

11
ultralytics/data/build.py
View file

@ -150,13 +150,12 @@ def check_source(source):
return source, webcam, screenshot, from_img, in_memory, tensor
def load_inference_source(source=None, imgsz=640, vid_stride=1, buffer=False):
def load_inference_source(source=None, vid_stride=1, buffer=False):
"""
Loads an inference source for object detection and applies necessary transformations.
Args:
source (str, Path, Tensor, PIL.Image, np.ndarray): The input source for inference.
imgsz (int, optional): The size of the image for inference. Default is 640.
vid_stride (int, optional): The frame interval for video sources. Default is 1.
buffer (bool, optional): Determined whether stream frames will be buffered. Default is False.
@ -172,13 +171,13 @@ def load_inference_source(source=None, imgsz=640, vid_stride=1, buffer=False):
elif in_memory:
dataset = source
elif webcam:
dataset = LoadStreams(source, imgsz=imgsz, vid_stride=vid_stride, buffer=buffer)
dataset = LoadStreams(source, vid_stride=vid_stride, buffer=buffer)
elif screenshot:
dataset = LoadScreenshots(source, imgsz=imgsz)
dataset = LoadScreenshots(source)
elif from_img:
dataset = LoadPilAndNumpy(source, imgsz=imgsz)
dataset = LoadPilAndNumpy(source)
else:
dataset = LoadImages(source, imgsz=imgsz, vid_stride=vid_stride)
dataset = LoadImages(source, vid_stride=vid_stride)
# Attach source types to the dataset
setattr(dataset, "source_type", source_type)