Skip to content

AXERA-TECH/CLIP-ONNX-AX650-CPP

Repository files navigation

CLIP

zh_clip-2023-10-20_10.19.10.mp4

Build with axcl

mkdir build
cd build
cmake -DCMAKE_TOOLCHAIN_FILE=../toolchains/aarch64-none-linux-gnu.toolchain.cmake -DOpenCV_DIR=${opencv_cmake_file_dir} -DBSP_MSP_DIR=/path/to/msp/out/ ..
make -j4

aarch64-none-gnu library:
opencv

Get model

cnclip clip

Run with axcl

# ./main --ienc cnclip/cnclip_vit_l14_336px_vision_u16u8.axmodel --tenc cnclip/cnclip_vit_l14_336px_text_u16.axmodel -i ../images/ -t text.txt -v ../cn_vocab.txt -l 1
[I][              load_image_encoder][  18]: input size 336 336
[I][              load_image_encoder][  22]: image feature len 768
[I][               load_text_encoder][  16]: text feature len 768
[I][                  load_tokenizer][  60]: text token len 52
[I][                   process_texts][  71]: encode text [bird] cost time : 0.004662
[I][                   process_texts][  71]: encode text [cat] cost time : 0.004529
[I][                   process_texts][  71]: encode text [dog] cost time : 0.004484
image encode cost time : 0.101203s
image encode cost time : 0.0918216s
image encode cost time : 0.094925s
postprocess cost time : 0.000195515s

per image:
                         image path\text|                                    bird|                                     cat|                                     dog|
                      ../images/bird.jpg|                                    1.00|                                    0.00|                                    0.00|
                       ../images/cat.jpg|                                    0.00|                                    1.00|                                    0.00|
                 ../images/dog-chai.jpeg|                                    0.00|                                    0.00|                                    1.00|


per text:
                         text\image path|                      ../images/bird.jpg|                       ../images/cat.jpg|                 ../images/dog-chai.jpeg|
                                    bird|                                    1.00|                                    0.00|                                    0.00|
                                     cat|                                    0.00|                                    1.00|                                    0.00|
                                     dog|                                    0.00|                                    0.01|                                    0.99|

Reference

CLIP
Chinese-CLIP
CLIP-ImageSearch-NCNN

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Packages

No packages published

Contributors 2

  •  
  •