Skip to content

Set DWPreprocessor.bbox_detector = yolox_l.torchscript.pt#1363

Merged
mcmonkey4eva merged 1 commit intomcmonkeyprojects:masterfrom
jtreminio:dwprocessor-bbox_detector
May 2, 2026
Merged

Set DWPreprocessor.bbox_detector = yolox_l.torchscript.pt#1363
mcmonkey4eva merged 1 commit intomcmonkeyprojects:masterfrom
jtreminio:dwprocessor-bbox_detector

Conversation

@jtreminio
Copy link
Copy Markdown
Contributor

@jtreminio jtreminio commented May 2, 2026

As noted in discord, I ran some benchmarks:

  • yolox_l.torchscript.pt = 7s
  • yolox_l.onnx = 51s
  • yolo_nas_l_fp16.onnx = 37s
  • yolo_nas_m_fp16.onnx = 32s
  • yolo_nas_s_fp16.onnx = 25s

All the onnx choices give an error, they require cuda 12:

onnx slower due to: [E:onnxruntime:Default, provider_bridge_ort.cc:2364 TryGetProviderInfo_CUDA] /onnxruntime_src/onnxruntime/core/session/provider_bridge_ort.cc:1957 onnxruntime::Provider& onnxruntime::ProviderLibrary::Get() [ONNXRuntimeError] : 1 : FAIL : Failed to load library /usr/local/lib/python3.12/dist-packages/onnxruntime/capi/libonnxruntime_providers_cuda.so with error: libcublasLt.so.12: cannot open shared object file: No such file or directory

Switching to yolox_l.torchscript.pt instead is safe because pose_estimator already uses dw-ll_ucoco_384_bs5.torchscript.pt so torchscript is assumed.

CleanShot 2026-05-01 at 19 02 17

@mcmonkey4eva mcmonkey4eva merged commit f856bbf into mcmonkeyprojects:master May 2, 2026
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants