Returns the mapping from model type name to Keras model.
()
| 39 | |
| 40 | |
| 41 | def get_models() -> Mapping[str, tf_keras.Model]: |
| 42 | """Returns the mapping from model type name to Keras model.""" |
| 43 | return { |
| 44 | 'efficientnet': efficientnet_model.EfficientNet.from_name, |
| 45 | 'resnet': resnet_model.resnet50, |
| 46 | 'vgg': vgg_model.vgg16, |
| 47 | } |
| 48 | |
| 49 | |
| 50 | def get_dtype_map() -> Mapping[str, tf.dtypes.DType]: |
no outgoing calls
no test coverage detected