Generates equal keras models with the desired data format
Project description
Keras data format converter
Generates equal keras models with the desired data format
Requirements
tensorflow >= 2.0
API
convert_channels_first_to_last(model: keras.Model, inputs_to_transpose: List[str] = None, verbose: bool = False) -> keras.Model
convert_channels_last_to_first(model: tf.keras.Model, inputs_to_transpose: List[str] = None, verbose: bool = False) \ -> tf.keras.Model
model
: Keras model to convert
inputs_to_transpose
: list of input names that need to be transposed due tothe data foramt changing
verbose
: detailed output
Getting started
from tensorflow import keras
from keras_data_format_converter import convert_channels_last_to_first
# Load Keras model
keras_model = keras.models.load_model("my_image_model")
# Call the converter (image_input is an input that needs to be transposed, can be different for your model)
converted_model = convert_channels_last_to_first(keras_model, ["image_input"])
Supported Layers with Special handling
- Normalization layers
- Permute
- Reshape
- Concatenate
- Dot
- MultiHeadAttention
- TFOpLambda (Inserted by the Functional API construction whenever users call a supported TF symbol on KerasTensors, see here at Tensorflow repo for more info)
Unsupported Layers due to lack of data_format property
- Cropping1D
- Upsampling1D
- Zeropadding1D
- All layers in tensorflow.keras.preprocessing
How to deploy
- Create a new release version on GitHub
- Update parameters in setup.py (usually
version
anddownload_url
) - Run
python setup.py sdist
in root directory - Run
pip install twine
- Run
twine upload dist/*
License
This software is covered by MIT License.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for keras-data-format-converter-0.0.9.dev2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 112dd764e0f01c331c5fae09b769ccc72f71d619a864b1de6d5611168f9c1fd8 |
|
MD5 | 72bcc1c65bf381627b751ad931e7c7d0 |
|
BLAKE2b-256 | 2dc6a34dda5a5a4485b94fd9e484bc3c01089158a0dcab9af822e2ed04acf71f |
Close
Hashes for keras_data_format_converter-0.0.9.dev2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3f50dc4c2d735f7f01d1fc2aac800ef8bd2b3750092dc77b9ecf66ee003b1099 |
|
MD5 | 0e9fe61af0becbc4a06f8f394ed03c72 |
|
BLAKE2b-256 | e56b54b1ef2a46aba8adf5e057a92e26c66acb7360d14f19c582905e9a817743 |