facenet模型转换

请问一下，facenet模型可以转换成功吗？我自己尝试去转换，显示转换成功了，但是推理的时候一直错误。下面是我转换的代码：

1. from rknn.api import RKNN
   
2. INPUT_SIZE = 160
   
3. 
   
4. if __name__ == '__main__':
   
5. 
   
6.     # Create RKNN object
   
7.     rknn = RKNN()
   
8. 
   
9.     # Config for Model Input PreProcess
   
10.     rknn.config(channel_mean_value='0 0 0 1', reorder_channel='0 1 2')
    
11.     print('config done')
    
12. 
    
13.     #load tensorflow model
    
14.     print('--> Loading model')
    
15.     rknn.load_tensorflow(tf_pb='./20180402-114759.pb',
    
16.                          inputs=['input'],
    
17.                          outputs=['InceptionResnetV1/Bottleneck/BatchNorm/Reshape_1'],                        
    
18.                          input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])                  
    
19.     print('done')
    
20. 
    
21.     # Build Model
    
22.     print('--> Building model')
    
23.     rknn.build(do_quantization=False)
    
24.     print('done')
    
25. 
    
26.     # Export RKNN Model
    
27.     rknn.export_rknn('./facenet2.rknn')
    
28. 
    
29.     rknn.release()        

复制代码

论坛上关于facenet转换的帖子太少，只好发帖求助了

ccgunner 发表于 2021-4-9 11:39
from rknn.api import RKNN

INPUT_SIZE = 160

多谢，我之前也用的这个，转换的时候会提示
“W Network may not support batch > 1 !”
你测试过转换后模型可以使用吗？