rockchip-linux / rknn-toolkit

BSD 3-Clause "New" or "Revised" License
771 stars 175 forks source link

Can not convert SwinTransformerV2 #441

Open pi-null-mezon opened 3 months ago

pi-null-mezon commented 3 months ago

I have failed to convert SwinTransformerV2 onnx (it works properly, I have tested) model into rknn.

rknn-toolkit versions I have tried: 1.7.3, 1.7.5

My script:

    ...
    # CONFIG
    rknn = RKNN(verbose=True, verbose_file=os.path.join(target_path, 'conversion.log'))
    rknn.config(quantize_input_node=True,
                mean_values=mean,
                std_values=std,
                quantized_dtype='asymmetric_affine-u8',            
                target_platform='rv1126',
                batch_size=100)

    # LOAD 
    ret = rknn.load_onnx(model=onnx_model_file, input_size_list=input_size,
                         inputs=inputs, outputs=outputs)  #..............................SUCCESS
    if ret != 0:
        print('Load ONNX model failed!')
        exit(ret)

    # BUILD RKNN MODEL
    ret = rknn.build(do_quantization=True,   
                     dataset=os.path.join(target_path, 'samples.txt'),
                     pre_compile=False)  #...............................................FAIL
   ... 

conversion.log [Uploading conversion.log…]()