Commit
·
9229863
1
Parent(s):
90d8dcc
fix onnx path
Browse files- detr/detr_models.py +1 -1
detr/detr_models.py
CHANGED
@@ -256,7 +256,7 @@ class SimpleDetrOnnx:
|
|
256 |
onnx_sess_opts.enable_mem_pattern = True
|
257 |
onnx_sess_opts.enable_cpu_mem_arena = True
|
258 |
self.ort_session = onnxruntime.InferenceSession(
|
259 |
-
f"{ONNX_DIR}/
|
260 |
sess_options=onnx_sess_opts,
|
261 |
providers=[
|
262 |
"CUDAExecutionProvider",
|
|
|
256 |
onnx_sess_opts.enable_mem_pattern = True
|
257 |
onnx_sess_opts.enable_cpu_mem_arena = True
|
258 |
self.ort_session = onnxruntime.InferenceSession(
|
259 |
+
f"{ONNX_DIR}/detr_simple_demo_onnx.onnx",
|
260 |
sess_options=onnx_sess_opts,
|
261 |
providers=[
|
262 |
"CUDAExecutionProvider",
|