MCPcopy
hub / github.com/PaddlePaddle/PaddleOCR / predict

Function predict

paddleocr-js/packages/core/src/models/det.ts:219–246  ·  view source on GitHub ↗
(cv, mats, overrides)

Source from the content-addressed store, hash-verified

217 return sessionState?.provider || "";
218 },
219 async predict(cv, mats, overrides) {
220 if (!sessionState?.session) {
221 throw new Error("Detection model session is not initialized.");
222 }
223 const params = resolveDetParams(defaultParams, overrides);
224 const batchSize = resolveRuntimeBatchSize(overrides?.batchSize, defaultBatchSize);
225 const results: DetResult[] = [];
226 const runCtx: DetRunContext = {
227 cv,
228 ort,
229 config,
230 session: sessionState.session
231 };
232 for (const chunk of chunkArray(mats, batchSize)) {
233 const preps = preprocess({ cv, ort, config }, chunk, params);
234 const inputTensor = packDetBatchTensor(ort, preps);
235 const fullOutput = await runInference(sessionState.session, inputTensor);
236 const internals = postprocess(runCtx, fullOutput, preps, params);
237 for (const internal of internals) {
238 results.push({
239 boxes: internal.boxes,
240 srcW: internal.prep.srcW,
241 srcH: internal.prep.srcH
242 });
243 }
244 }
245 return results;
246 },
247 async dispose() {
248 await releaseSessions(sessionState?.session);
249 sessionState = null;

Callers

nothing calls this directly

Calls 7

resolveRuntimeBatchSizeFunction · 0.90
chunkArrayFunction · 0.90
runInferenceFunction · 0.90
resolveDetParamsFunction · 0.85
packDetBatchTensorFunction · 0.85
preprocessFunction · 0.70
postprocessFunction · 0.70

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…