Spaces:
Build error
Build error
Jakub Kwiatkowski
commited on
Commit
·
76b576d
1
Parent(s):
44b7335
fix: Fix figure order in render_panel.
Browse files
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
|