CuriousDolphin commited on
Commit
9229863
·
1 Parent(s): 90d8dcc

fix onnx path

Browse files
Files changed (1) hide show
  1. 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}/detr_simple_demo.onnx",
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",