search

broadinstitute / keras-rcnn

Keras package for region-based convolutional neural networks (RCNNs)
Other
553 stars 224 forks source link

anchor() issue #111

Open dmitrysarov opened 6 years ago

dmitrysarov commented 6 years ago

from keras_rcnn.backend.common import * import tensorflow as tf sess = tf.InteractiveSession() anc = anchor(4,[1,2],[1])

[[ 0. 0. 3. 3. ] [ 0. 0. 3. 3. ] [ 0.5 -1. 2.5 4. ] [ 0.5 -1. 2.5 4. ]]

I suppose that it should return list of two arrays.

dmitrysarov commented 6 years ago

As solution, i guess could be add to def _scale_enum(anchor, scales):

such strings

x_ctr = keras.backend.expand_dims(x_ctr, 1)
y_ctr = keras.backend.expand_dims(y_ctr, 1)