Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
| import urllib.parse | |
| import pandas as pd | |
| from gradio.themes.utils.colors import Color | |
| DATA_URL = 'https://raw.githubusercontent.com/Deeplite/deeplite-torch-zoo/develop/results/yolobench/' | |
| DEEPLITE_DARK_BLUE_RGB = (0, 66, 107) | |
| DEEPLITE_DARK_BLUE_HEX = '#00426B' | |
| DEEPLITE_LIGHT_BLUE_RGB = (0, 148, 206) | |
| DEEPLITE_LIGHT_BLUE_HEX = '#0094CE' | |
| DEEPLITE_DARK_BLUE_GRADIO = Color( | |
| name='deeplite_dark_blue', | |
| c50=DEEPLITE_DARK_BLUE_HEX, | |
| c100=DEEPLITE_DARK_BLUE_HEX, | |
| c200=DEEPLITE_DARK_BLUE_HEX, | |
| c300=DEEPLITE_DARK_BLUE_HEX, | |
| c400=DEEPLITE_DARK_BLUE_HEX, | |
| c500=DEEPLITE_DARK_BLUE_HEX, | |
| c600=DEEPLITE_DARK_BLUE_HEX, | |
| c700=DEEPLITE_DARK_BLUE_HEX, | |
| c800=DEEPLITE_DARK_BLUE_HEX, | |
| c900=DEEPLITE_DARK_BLUE_HEX, | |
| c950=DEEPLITE_DARK_BLUE_HEX, | |
| ) | |
| DEEPLITE_LIGHT_BLUE_GRADIO = Color( | |
| name='deeplite_dark_blue', | |
| c50=DEEPLITE_LIGHT_BLUE_HEX, | |
| c100=DEEPLITE_LIGHT_BLUE_HEX, | |
| c200=DEEPLITE_LIGHT_BLUE_HEX, | |
| c300=DEEPLITE_LIGHT_BLUE_HEX, | |
| c400=DEEPLITE_LIGHT_BLUE_HEX, | |
| c500=DEEPLITE_LIGHT_BLUE_HEX, | |
| c600=DEEPLITE_LIGHT_BLUE_HEX, | |
| c700=DEEPLITE_LIGHT_BLUE_HEX, | |
| c800=DEEPLITE_LIGHT_BLUE_HEX, | |
| c900=DEEPLITE_LIGHT_BLUE_HEX, | |
| c950=DEEPLITE_LIGHT_BLUE_HEX, | |
| ) | |
| def load_yolobench_data(): | |
| df = pd.read_csv(urllib.parse.urljoin(DATA_URL, 'merged_results.csv')) | |
| pareto_indices_df = pd.read_csv(urllib.parse.urljoin(DATA_URL, 'pareto_indices.csv')) | |
| pareto_indices = {} | |
| for row_idx in range(pareto_indices_df.shape[0]): | |
| data_key = pareto_indices_df.iloc[row_idx, :]['data'] | |
| if data_key not in pareto_indices: | |
| pareto_indices[data_key] = {} | |
| hw_key = pareto_indices_df.iloc[row_idx, :]['hardware'] | |
| indices = pareto_indices_df.iloc[row_idx, :]['pareto_indices'] | |
| indices = [int(val) for val in indices.split(',')] | |
| pareto_indices[data_key][hw_key] = indices | |
| return df, pareto_indices | |