talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve docstring of image classifier model spec.

Browse Source 
PiperOrigin-RevId: 515466722
This commit is contained in:
MediaPipe Team
2023-03-09 16:06:40 -08:00
committed by Copybara-Service
parent 517e997179
commit 2d8f937913
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/model_maker/python/vision/image_classifier/model_spec.py
+1 -1
View File
@@ -28,7 +28,7 @@ class ModelSpec(object):
uri: str, uri: str,
input_image_shape: Optional[List[int]] = None, input_image_shape: Optional[List[int]] = None,
name: str = ''): name: str = ''):
"""Initializes a new instance of the `ImageModelSpec` class. """Initializes a new instance of the image classifier `ModelSpec` class.
Args: Args:
uri: str, URI to the pretrained model. uri: str, URI to the pretrained model.