deeplearning4j: Unsupported space_to_batch operation in Lambda layer for YOLO import
Issue Description
While trying to import a Keras model for YOLO created via YAD2K, I get an “Unsupported keras layer type Lambda” error for this layer:
space_to_depth_x2 (Lambda) (None, 19, 19, 256) 0 leaky_re_lu_21[0][0]
It corresponds to https://www.tensorflow.org/api_docs/python/tf/space_to_depth in TensorFlow. It does not look like a complicated operation, but do you know if we have a layer like that somewhere in DL4J? What do you think @AlexDBlack?
Version Information
- Deeplearning4j 0.9.2-SNAPSHOT
About this issue
- Original URL
- State: closed
- Created 7 years ago
- Comments: 17 (13 by maintainers)
Commits related to this issue
- Keras YOLO test, for #4256 — committed to deeplearning4j/deeplearning4j by maxpumperla 6 years ago
I’m renaming this issue to this more specific case and open a new “Lambda layer support” issue. thanks!
@maxpumperla I was able to reproduce the issue with that one ? Its size is 36kb. my_model.zip
I can confirm that from a business point of view, Tiny Yolo is not accurate enough for many use-cases. I work on a proof of concept trying to convince my partners that java and dl4j are viable compared to the classical python approach. One strong argument is “Keras can be imported” but today state of the art objects detection model Yolo cannot be imported. I really hope that this capability will be added soon.