Skip to content

Commit

Permalink
move out stuff from __init__
Browse files Browse the repository at this point in the history
  • Loading branch information
Mikel Broström committed Jul 8, 2024
1 parent f0cf71f commit d4fa0e9
Showing 1 changed file with 0 additions and 17 deletions.
17 changes: 0 additions & 17 deletions boxmot/appearance/reid_model_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,23 +123,6 @@
}


# Combine into a single data structure
combined_data = {}


for model_type in __model_types:
for url_key, url_value in __trained_urls.items():
if model_type in url_key:
dataset = next(key for key in NR_CLASSES_DICT.keys() if key in url_key)
num_classes = NR_CLASSES_DICT[dataset]
combined_data[model_type] = {
'trained_url': url_value,
'dataset': dataset,
'num_classes': num_classes
}
break # Stop searching further once found
print(combined_data)

# Utility functions
def show_downloadable_models():
LOGGER.info("Available .pt ReID models for automatic download")
Expand Down

0 comments on commit d4fa0e9

Please sign in to comment.