|
25 | 25 | using json = nlohmann::json;
|
26 | 26 |
|
27 | 27 | std::string DATA_DIR = "../data";
|
28 |
| -std::string MODEL_PATH_TEMPLATE = "public/%s/FP16/%s.xml"; |
| 28 | +std::string MODEL_PATH_TEMPLATE = "otx_models/%s.xml"; |
29 | 29 | std::string IMAGE_PATH = "coco128/images/train2017/000000000074.jpg";
|
30 | 30 |
|
31 | 31 | std::string TMP_MODEL_FILE = "tmp_model.xml";
|
@@ -115,6 +115,7 @@ TEST_P(ClassificationModelParameterizedTestSaveLoad, TestClassificationCorrectne
|
115 | 115 | }
|
116 | 116 |
|
117 | 117 | auto model_path = string_format(MODEL_PATH_TEMPLATE, GetParam().name.c_str(), GetParam().name.c_str());
|
| 118 | + std::cout << model_path << "\n"; |
118 | 119 | bool preload = true;
|
119 | 120 | auto model = ClassificationModel::create_model(DATA_DIR + "/" + model_path, {}, preload, "CPU");
|
120 | 121 |
|
@@ -260,16 +261,16 @@ TEST_P(DetectionModelParameterizedTestSaveLoad, TestDetctionCorrectnessAfterSave
|
260 | 261 |
|
261 | 262 | INSTANTIATE_TEST_SUITE_P(ClassificationTestInstance,
|
262 | 263 | ClassificationModelParameterizedTest,
|
263 |
| - ::testing::Values(ModelData("efficientnet-b0-pytorch"))); |
| 264 | + ::testing::Values(ModelData("mlc_mobilenetv3_large_voc"))); |
264 | 265 | INSTANTIATE_TEST_SUITE_P(ClassificationTestInstance,
|
265 | 266 | ClassificationModelParameterizedTestSaveLoad,
|
266 |
| - ::testing::Values(ModelData("efficientnet-b0-pytorch"))); |
| 267 | + ::testing::Values(ModelData("mlc_mobilenetv3_large_voc"))); |
267 | 268 | INSTANTIATE_TEST_SUITE_P(SSDTestInstance,
|
268 | 269 | SSDModelParameterizedTest,
|
269 |
| - ::testing::Values(ModelData("ssdlite_mobilenet_v2"), ModelData("ssd_mobilenet_v1_fpn_coco"))); |
| 270 | + ::testing::Values(ModelData("detection_model_with_xai_head"))); |
270 | 271 | INSTANTIATE_TEST_SUITE_P(SSDTestInstance,
|
271 | 272 | DetectionModelParameterizedTestSaveLoad,
|
272 |
| - ::testing::Values(ModelData("ssdlite_mobilenet_v2"), ModelData("ssd_mobilenet_v1_fpn_coco"))); |
| 273 | + ::testing::Values(ModelData("detection_model_with_xai_head"))); |
273 | 274 |
|
274 | 275 | class InputParser {
|
275 | 276 | public:
|
|
0 commit comments