cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests 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 15:59:45 -08:00 committed by Copybara-Service
parent 517e997179
commit 2d8f937913
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/model_maker/python/vision/image_classifier/model_spec.py
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.