5 lines
241 B
Python
5 lines
241 B
Python
from onnxruntime import InferenceSession
|
|
|
|
PADDLE_DET = InferenceSession("model/object_det_model/ppyoloe_plus_crn_l_80e_coco_w_nms.onnx",
|
|
providers=["CUDAExecutionProvider"], provider_options=[{"device_id": 0}])
|