Jakub Kwiatkowski commited on
Commit
76b576d
·
1 Parent(s): 44b7335

fix: Fix figure order in render_panel.

Browse files
Files changed (1) hide show
  1. raven_utils/render/rendering.py +13 -0
raven_utils/render/rendering.py CHANGED
@@ -82,6 +82,19 @@ def render_panels(feature, target=True, angle=None):
82
  # rv.decode_output_reshape(root)
83
  # decoded =
84
  # panel = decoded[0]
 
 
 
 
 
 
 
 
 
 
 
 
 
85
  panels = []
86
  for group, exist, color, size, type_ in zip(*raven_utils.decode.decode_target_flat(feature)):
87
  canvas = np.ones((IMAGE_SIZE, IMAGE_SIZE), np.uint8) * 255
 
82
  # rv.decode_output_reshape(root)
83
  # decoded =
84
  # panel = decoded[0]
85
+
86
+ # hack due to different order for in_4_out_1
87
+ feature = np.concatenate(
88
+ [
89
+ feature[:, :74],
90
+ feature[:, 86:89],
91
+ feature[:, 77:86],
92
+ feature[:, 74:77],
93
+ feature[:, 89:]
94
+ ],
95
+ axis=-1
96
+ )
97
+
98
  panels = []
99
  for group, exist, color, size, type_ in zip(*raven_utils.decode.decode_target_flat(feature)):
100
  canvas = np.ones((IMAGE_SIZE, IMAGE_SIZE), np.uint8) * 255