Skip to content

Commit 3457737

Browse files
jw098Gin890
authored andcommitted
minor
1 parent 5125fb9 commit 3457737

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

SerialPrograms/Source/ML/DataLabeling/ML_SegmentAnythingModel.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,7 @@ void compute_embeddings_for_folder(const std::string& embedding_model_path, cons
242242
for(size_t j = 0; j < 2; j++){
243243
try{
244244
// If fails with GPU, fall back to CPU.
245+
// throw Ort::Exception("Testing.", ORT_FAIL); // to simulate GPU/CPU failure
245246
embedding_session->run(resized_mat, output_image_embedding);
246247
break;
247248
}catch(Ort::Exception& e){
@@ -257,7 +258,7 @@ void compute_embeddings_for_folder(const std::string& embedding_model_path, cons
257258
return;
258259
}
259260
}catch(...){
260-
std::cerr << "Error: Unknown error." << std::endl;
261+
std::cerr << "Error: Unknown error. Embedding session failed." << std::endl;
261262
QMessageBox box;
262263
box.warning(nullptr, "Error:",
263264
QString::fromStdString("Error: Unknown error. Embedding session failed."));

SerialPrograms/Source/ML/Inference/ML_YOLOv5Detector.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,11 +106,9 @@ bool YOLOv5Detector::detect(const ImageViewRGB32& screen){
106106
std::vector<std::string> labels = m_yolo_session->get_label_names();
107107
m_yolo_session = std::make_unique<YOLOv5Session>(m_model_path, std::move(labels), m_use_gpu);
108108
}else{
109-
std::cerr << "Error: YOLO session failed even when using the CPU.\n" << e.what() << std::endl;
110-
throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, "Error: YOLO session failed.");
109+
throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, "Error: YOLO session failed even when using the CPU." + std::string(e.what()));
111110
}
112111
}catch(...){
113-
std::cerr << "Error: Unknown error." << std::endl;
114112
throw InternalProgramError(nullptr, PA_CURRENT_FUNCTION, "Unknown error: YOLO session failed.");
115113

116114
}

0 commit comments

Comments
 (0)