rknn.load_tensorflow() 时报错 segmentation fault

代码

1. rknn = RKNN(verbose=True)
   
2. ret = rknn.init_runtime(target='rk3399pro')
   
3. if ret != 0:
   
4.     print('Init runtime environment failed')
   
5.     exit(ret)
   
6. print('done')
   
7. 
   
8. rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2')
   
9. print('--> Loading model')
   
10. rknn.load_tensorflow(tf_pb='XXXXXXXX.pb',
    
11.                      inputs=['input_image'],
    
12.                      outputs=['XXXXXX1', 'XXXXXX2', 'XXXXXX3', 'XXXXXX4', 'XXXXXX5', 'XXXXXX6', ],
    
13.                      input_size_list=[[512, 768, 3]])
    
14. print('done')
    
15. 
    

复制代码


输出

1. 
   
2. --> Init runtime environment
   
3. done
   
4. --> Loading model
   
5. [1]    28312 segmentation fault (core dumped)  ./convert.py
   
6. 
   

复制代码


即使开了verbose，也没有得到任何有用的信息。
我把这个贴出来好像也没啥用，有啥方法可以更verbose一点吗？
或者说遇到 segmentation fault 应该去检查哪里呢？

可以使用最新的0.9.8版本试一下

就是0.9.8。

1. init_runtime调用的位置不对
2. 你这边版本的toolkit是从其他版本升级上来的吗？如果是，使用全新的环境试试。

1. “init_runtime调用的位置不对”具体指的是什么？我这边代码基本抄的 http://t.rock-chips.com/wiki.php?mod=view&id=36
2. 之前是RKNN-Toolkit 0.9.7升到0.9.8的，刚刚用干净环境安装RKNN-Toolkit 0.9.8，问题依旧。

有什么方法能够多打印一点debug信息出来吗？

init_runtime应该在build或load_rknn完成之后调用，参考

1. if __name__ == '__main__':
   
2. 
   
3.     # Create RKNN object
   
4.     rknn = RKNN()
   
5. 
   
6.     # Config for Model Input PreProcess
   
7.     rknn.config(channel_mean_value='128 128 128 128', reorder_channel='0 1 2', batch_size=5)
   
8. 
   
9.     # Load TensorFlow Model
   
10.     print('--> Loading model')
    
11.     rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v1_coco_2017_11_17.pb',
    
12.                          inputs=['FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/batchnorm/mul_1'],
    
13.                          outputs=['concat', 'concat_1'],
    
14.                          input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
    
15.     print('done')
    
16. 
    
17.     # Build Model
    
18.     print('--> Building model')
    
19.     rknn.build(do_quantization=True, dataset='./dataset.txt')
    
20.     print('done')
    
21. 
    
22.     # Export RKNN Model
    
23.     rknn.export_rknn('./ssd_mobilenet_v1_coco.rknn')
    
24. 
    
25.     # Direct Load RKNN Model
    
26.     # rknn.load_rknn('./ssd_mobilenet_v1_coco.rknn')
    
27. 
    
28.     # Set inputs
    
29.     orig_img = cv2.imread('./road.bmp')
    
30.     img = cv2.cvtColor(orig_img, cv2.COLOR_BGR2RGB)
    
31.     img = cv2.resize(img, (INPUT_SIZE, INPUT_SIZE), interpolation=cv2.INTER_CUBIC)
    
32. 
    
33.     # init runtime environment
    
34.     print('--> Init runtime environment')
    
35.     ret = rknn.init_runtime()
    
36.     if ret != 0:
    
37.         print('Init runtime environment failed')
    
38.         exit(ret)
    
39.     print('done')
    
40. 
    
41.     # Inference
    
42.     print('--> Running model')
    
43.     outputs = rknn.inference(inputs=[img])
    
44.     print('done')
    
45.     print('inference result: ', outputs)

复制代码

后置了rknn.init_runtime()，问题依旧。

你是在pc上运行还是在3399pro上呢，example里面的例子可以正常运行吗

在RK3399Pro上面运行。
三个example可以正常运行。
自己做了个简化版的模型可以正常运行。
完整版模型在rknn.load_tensorflow()的时候segmentation fault。