Skip to content

Commit 1c0f085

Browse files
committed
remove obsolete load
1 parent d7d5199 commit 1c0f085

2 files changed

Lines changed: 1 addition & 7 deletions

File tree

src/yolo11segncnn.cpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,6 @@ YOLOv11SegDetectorNcnn::YOLOv11SegDetectorNcnn(const std::string &modelPath,
6666

6767
outputNames = net.output_names();
6868

69-
classNames = utils::getClassNames(labelsPath);
70-
classColors = utils::generateColors(classNames);
71-
7269
qDebug() << "[INFO] YOLOv11Seg loaded: " << modelPath;
7370
qDebug() << " Input shape: " << inputImageShape.height << "x" << inputImageShape.width
7471
<< (isDynamicInputShape ? " (dynamic)" : "");

src/yolo11segonnx.cpp

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ YOLOv11SegDetectorOnnx::YOLOv11SegDetectorOnnx(const std::string &modelPath,
1111
bool useGPU)
1212
: Yolo11Segementation(labelsPath), env(ORT_LOGGING_LEVEL_WARNING, "YOLOv11Seg")
1313
{
14-
sessionOptions.SetIntraOpNumThreads(std::min(6, static_cast<int>(std::thread::hardware_concurrency())));
14+
sessionOptions.SetIntraOpNumThreads(std::min(8, static_cast<int>(std::thread::hardware_concurrency())));
1515
sessionOptions.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_ALL);
1616

1717
std::vector<std::string> providers = Ort::GetAvailableProviders();
@@ -87,9 +87,6 @@ YOLOv11SegDetectorOnnx::YOLOv11SegDetectorOnnx(const std::string &modelPath,
8787
outputNames.push_back(outputNameAllocs.back().get());
8888
}
8989

90-
classNames = utils::getClassNames(labelsPath);
91-
classColors = utils::generateColors(classNames);
92-
9390
qDebug() << "[INFO] YOLOv11Seg loaded: " << modelPath;
9491
qDebug() << " Input shape: " << inputImageShape.height << "x" << inputImageShape.width
9592
<< (isDynamicInputShape ? " (dynamic)" : "");

0 commit comments

Comments
 (0)