SHARE
TWEET

Untitled

a guest Jul 17th, 2019 56 Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. diff --git a/net/bbox_3D_net.py b/net/bbox_3D_net.py
  2. index 6daafdd..8146caa 100644
  3. --- a/net/bbox_3D_net.py
  4. +++ b/net/bbox_3D_net.py
  5. @@ -18,7 +18,7 @@ def bbox_3D_net(input_shape=(224, 224, 3), vgg_weights=None, freeze_vgg=False, b
  6.          for layer in vgg16_model.layers:
  7.              layer.trainable = False
  8.  
  9. -    x = Flatten()(vgg16_model.output)
  10. +    x = Reshape((25088,))(vgg16_model.output)
  11.  
  12.      dimension = Dense(512)(x)
  13.      dimension = LeakyReLU(alpha=0.1)(dimension)
  14. @@ -31,7 +31,7 @@ def bbox_3D_net(input_shape=(224, 224, 3), vgg_weights=None, freeze_vgg=False, b
  15.      orientation = Dropout(0.5)(orientation)
  16.      orientation = Dense(bin_num * 2)(orientation)
  17.      orientation = LeakyReLU(alpha=0.1)(orientation)
  18. -    orientation = Reshape((bin_num, -1))(orientation)
  19. +    orientation = Reshape((bin_num, 2))(orientation)
  20.      orientation = Lambda(l2_normalize, name='orientation')(orientation)
  21.  
  22.      confidence = Dense(256)(x)
RAW Paste Data
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy. OK, I Understand
 
Top