fix int type

open-mmlab · Jul 17, 2023 · cd11954 · cd11954
1 parent 74305ce
commit cd11954
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/csrc/mmdeploy/net/rknn/rknn_net.cpp b/csrc/mmdeploy/net/rknn/rknn_net.cpp
@@ -170,7 +170,7 @@ Result<void> RKNNNet::Init(const Value& args) {
       MMDEPLOY_ERROR("rknn query 'RKNN_QUERY_INPUT_ATTR' fail! ret: {}", ret);
       return Status(eFail);
     }
-    if (attr.type != RKNN_TENSOR_UINT8) {
+    if (attr.type != RKNN_TENSOR_UINT8 || attr.type != RKNN_TENSOR_INT8) {
       MMDEPLOY_ERROR("MMDeploy SDK only supports RKNN-INT8 model");
       return Status(eInvalidArgument);
     }

diff --git a/docs/zh_cn/01-how-to-build/rockchip.md b/docs/zh_cn/01-how-to-build/rockchip.md
@@ -130,8 +130,13 @@ git clone https://github.com/open-mmlab/mmdetection
 
 # 执行转换命令
 python tools/deploy.py \
-    configs/mmpretrain/detection_rknn_static.py \
-
+    configs/mmdet/detection/detection_rknn-int8_static-320x320.py \
+    /the/path/of/mmdet/configs/yolov3/yolov3_mobilenetv2_320_300e_coco.py \
+    https://download.openmmlab.com/mmdetection/v2.0/yolo/yolov3_mobilenetv2_320_300e_coco/yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth \
+    /the/path/of/mmdet/demo/demo.jpg \
+    --work-dir mmdeploy_models/mmdet/yolov3 \
+    --device cpu \
+    --dump-info
 ```
 
 - RTMDet