Skip to content

Support for More Layers #13

Open
Open
@frytoli

Description

@frytoli

Firstly, nice work on this project -- this looks really cool!

I'm working on a computer vision project, and would love to use Konverter to convert some pre-trained models. As such, these models include ZeroPadding2D layers. Because this layer type isn't supported, when I run Konverter on the model (yes I tried anyway haha) this is the resulting traceback and error:

Traceback (most recent call last):
File "test.py", line 14, in
konverter.konvert(os.path.join(os.getcwd(), 'models_eynollah', 'model_page_mixed_best.h5'), output_file=os.path.join(os.getcwd(), 'models', 'page_mixed_best.py'))
File "/venv/lib/python3.7/site-packages/konverter/init.py", line 20, in init
no_watermark=no_watermark)
File "/venv/lib/python3.7/site-packages/konverter/init.py", line 41, in _konvert
self.start()
File "/venv/lib/python3.7/site-packages/konverter/init.py", line 45, in start
self.get_layers()
File "/venv/lib/python3.7/site-packages/konverter/init.py", line 214, in get_layers
layer = support.get_layer_info(layer, next_layer)
File "/venv/lib/python3.7/site-packages/konverter/utils/konverter_support.py", line 104, in get_layer_info
layer_class.info.activation = self._get_layer_activation(layer)
File "/venv/lib/python3.7/site-packages/konverter/utils/konverter_support.py", line 80, in _get_layer_activation
if hasattr(layer.activation, '_keras_api_names'):
AttributeError: 'ZeroPadding2D' object has no attribute 'activation'

Do you have any advice for how I could add functionality for more layers and/or would you consider adding support for more layers?

Thanks!

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions