add rknn device check

mmdeploy/apis/rknn/__init__.py

@@ -1,7 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-from mmdeploy.backend.rknn import is_available
+from mmdeploy.backend.rknn import device_available, is_available, package_info
 
-__all__ = ['is_available']
+__all__ = ['is_available', 'device_available', 'package_info']
 
 if is_available():
     from mmdeploy.backend.rknn.onnx2rknn import onnx2rknn as _onnx2rknn

mmdeploy/backend/rknn/__init__.py

@@ -13,15 +13,26 @@ def is_available():
     return importlib.util.find_spec('rknn') is not None
 
 
+def package_info():
+    import pkg_resources
+    for p in pkg_resources.working_set:
+        if p.project_name.startswith('rknn-toolkit'):
+            return dict(name=p.project_name, version=p.version)
+    return dict(name=None, version=None)
+
+
 def device_available():
     """Check whether device available.
 
     Returns:
         bool: True if the device is available.
     """
-    ret = subprocess.check_output('adb devices', shell=True)
-    match = re.search(r'\\n\w+\\tdevice', str(ret))
-    return match is not None
+    try:
+        ret = subprocess.check_output('adb devices', shell=True)
+        match = re.search(r'\\n\w+\\tdevice', str(ret))
+        return match is not None
+    except Exception:
+        return False
 
 
 __all__ = []

tools/check_env.py

@@ -35,6 +35,10 @@ def check_backend():
     logger.info(f'ncnn: {ncnn_version}\tops_is_avaliable : '
                 f'{ncnn_apis.is_custom_ops_available()}')
 
+    import mmdeploy.apis.rknn as rknn_apis
+    logger.info(f'rknn: {rknn_apis.package_info()}\t'
+                f'adb_device_available: {rknn_apis.device_available()}')
+
     import mmdeploy.apis.pplnn as pplnn_apis
     logger.info(f'pplnn_is_avaliable: {pplnn_apis.is_available()}')