|
|
|
@@ -28,7 +28,7 @@ class InputType(object): |
|
|
|
INPUT_TYPE = { |
|
|
|
InputType.IMAGE: (str, np.ndarray, Image.Image), |
|
|
|
InputType.TEXT: str, |
|
|
|
InputType.AUDIO: (str, np.ndarray), |
|
|
|
InputType.AUDIO: (str, bytes, np.ndarray), |
|
|
|
InputType.VIDEO: (str, np.ndarray, cv2.VideoCapture), |
|
|
|
InputType.BOX: (list, np.ndarray), |
|
|
|
InputType.DICT: (dict, type(None)), |
|
|
|
|