|
|
@@ -15,7 +15,7 @@ SAFETY_DETECTION_CONFIG = {
|
|
|
# - YOLO: .pt 文件, 使用 ultralytics
|
|
|
# - RKNN: .rknn 文件, 使用 rknnlite (RK3588 平台)
|
|
|
# - ONNX: .onnx 文件, 使用 onnxruntime
|
|
|
- 'model_path': '/Users/wenhongquan/Desktop/阿里云同步/项目/dnn/德胜河 AI/dsh/testrk3588/yolo11n_rk3588.pt',
|
|
|
+ 'model_path': '/home/admin/dsh/testrk3588/yolo11n_rk3588.pt',
|
|
|
|
|
|
'model_type': 'yolo', # 模型类型: 'auto', 'yolo', 'rknn', 'onnx'
|
|
|
'use_gpu': False, # RKNN 使用 NPU,不依赖 GPU
|