Spaces:
Build error
Build error
Jakub Kwiatkowski
commited on
Commit
·
2526f4b
1
Parent(s):
38f87b5
Refactor.
Browse files
raven_utils/render/rendering.py
CHANGED
@@ -86,7 +86,7 @@ def render_panels(feature, target=True,angle=None):
|
|
86 |
# decoded =
|
87 |
# panel = decoded[0]
|
88 |
panels = []
|
89 |
-
for group, exist, color, size, type_ in
|
90 |
canvas = np.ones((IMAGE_SIZE, IMAGE_SIZE), np.uint8) * 255
|
91 |
structure_img = render_structure(group)
|
92 |
background = np.zeros((IMAGE_SIZE, IMAGE_SIZE), np.uint8)
|
|
|
86 |
# decoded =
|
87 |
# panel = decoded[0]
|
88 |
panels = []
|
89 |
+
for group, exist, color, size, type_ in zip(*raven_utils.decode.decode_target_flat(feature)):
|
90 |
canvas = np.ones((IMAGE_SIZE, IMAGE_SIZE), np.uint8) * 255
|
91 |
structure_img = render_structure(group)
|
92 |
background = np.zeros((IMAGE_SIZE, IMAGE_SIZE), np.uint8)
|