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

Refactor with simplifications (#19329)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2025-02-20 19:43:53 +08:00 • committed by GitHub
parent f4307339ad
commit 82b1ce44cc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 14 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

10
examples/YOLOv8-Segmentation-ONNXRuntime-Python/main.py
View file

@ -79,9 +79,7 @@ class YOLOv8Seg:
predictions = self.session.run(None, {self.session.get_inputs()[0].name: processed_image})
# Post-process
results = self.postprocess(im0, processed_image, predictions)
return results
return self.postprocess(im0, processed_image, predictions)
def preprocess(self, image, new_shape: Union[Tuple, List] = (640, 640)):
"""
@ -99,8 +97,7 @@ class YOLOv8Seg:
"""
image, _, _ = self.__resize_and_pad_image(image=image, new_shape=new_shape)
image = self.__reshape_image(image=image)
processed_image = image[None] if len(image.shape) == 3 else image
return processed_image
return image[None] if len(image.shape) == 3 else image
def __reshape_image(self, image: np.ndarray) -> np.ndarray:
"""
@ -117,8 +114,7 @@ class YOLOv8Seg:
"""
image = image.transpose([2, 0, 1])
image = image[np.newaxis, ...]
image = np.ascontiguousarray(image).astype(np.float32) / 255
return image
return np.ascontiguousarray(image).astype(np.float32) / 255
def __resize_and_pad_image(
self, image=np.ndarray, new_shape: Union[Tuple, List] = (640, 640), color: Union[Tuple, List] = (114, 114, 114)