| # /*--------------------------------------------------------------------------------------------- | |
| # * Copyright (c) 2025 STMicroelectronics. | |
| # * All rights reserved. | |
| # * | |
| # * This software is licensed under terms that can be found in the LICENSE file in | |
| # * the root directory of this software component. | |
| # * If no LICENSE file comes with this software, it is provided AS-IS. | |
| # *--------------------------------------------------------------------------------------------*/ | |
| import os | |
| CHECKPOINT_STORAGE_URL = "https://github.com/STMicroelectronics/stm32ai-modelzoo/tree/main/image_classification/torch_checkpoints/" | |
| CURRENT_REPO_PATH = os.path.dirname(os.path.abspath(__file__)) | |
| SERVICES_ROOT = os.path.abspath(os.path.join(CURRENT_REPO_PATH, '../../../..')) | |
| #CHECKPOINT_STORAGE_URL = os.path.join(os.path.dirname(SERVICES_ROOT), "stm32ai-modelzoo/image_classification/torch_checkpoints/") | |
| # TODO : add more models here with time, library based models trained on food, flowers, and custom models | |
| # trained on food, flowers, imagenet | |
| MODEL_CHECKPOINTS = { | |
| "st_resnettiny_actrelu_pt_datasetimagenet_res224" : "st_resnettiny_actrelu_pt_224.pth.tar", | |
| "st_resnetmilli_actrelu_pt_datasetimagenet_res224" : "st_resnetmilli_actrelu_pt_224.pth.tar", | |
| "st_resnetmicro_actrelu_pt_datasetimagenet_res224" : "st_resnetmicro_actrelu_pt_224.pth.tar", | |
| "st_resnetnano_actrelu_pt_datasetimagenet_res224" : "st_resnetnano_actrelu_pt_224.pth.tar", | |
| "st_resnetpico_actrelu_pt_datasetimagenet_res224" : "st_resnetpico_actrelu_pt_224.pth.tar", | |
| "mobilenetv2_w035_pt_datasetimagenet_res224": "mobilenetv2_w035_pt_224.pt", | |
| # "mobilenet_v1_pt_datasetvww_res224": "mobilenetv1-vww-84f65dc4bc649cd6.pth", | |
| # "mobilenet_v1_0.25_pt_datasetvww_res224": "mobilenet_v1_0.25.pt", | |
| # "mobilenet_v1_0.25_96px_pt_datasetvww_res96": "mobilenet_v1_0.25_96px_798-8df2181bdab1433e.pt" | |
| } |