Skip to content

Commit

Permalink
fix colors and coords
Browse files Browse the repository at this point in the history
  • Loading branch information
almazgimaev committed Dec 17, 2024
1 parent 8443c72 commit 0a5f9ac
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -383,8 +383,13 @@ def run():
obj_classes = list(set(all_info["object_cls"]))
print(f"n_classes = {len(obj_classes)}")
series, colors, global_idxs_mapping = run_utils.make_series(projections, all_info_list, project_meta)
x_coordinates = [i["x"] for i in series[0]["data"]]
y_coordinates = [i["y"] for i in series[0]["data"]]

x_coordinates, y_coordinates, colors = [], [], []
for s, color in zip(series, colors):
x_coordinates.extend([i["x"] for i in s["data"]])
y_coordinates.extend([i["y"] for i in s["data"]])
colors.extend([color] * len(s["data"]))

plot = Bokeh.Circle(x_coordinates, y_coordinates, radii=0.05, colors=colors)
bokeh.add_plots([plot])
bokeh_iframe.set(bokeh.get_html_route_with_timestamp(), height="600px", width="100%")
Expand Down

0 comments on commit 0a5f9ac

Please sign in to comment.