search

axinc-ai / ailia-models

The collection of pre-trained, state-of-the-art AI models for ailia SDK
2.04k stars 325 forks source link

Implement Florence-2 #1561

Closed ooe1123 closed 1 month ago

ooe1123 commented 1 month ago

https://github.com/axinc-ai/ailia-models/issues/1541 のPRです

kyakuno commented 1 month ago

ONNX Runtime (CPU)

 INFO florence2.py (517) :  ailia processing estimation time 4146 ms
 INFO florence2.py (517) :  ailia processing estimation time 4032 ms
 INFO florence2.py (517) :  ailia processing estimation time 4667 ms
 INFO florence2.py (517) :  ailia processing estimation time 4445 ms
 INFO florence2.py (517) :  ailia processing estimation time 5450 ms
 INFO florence2.py (521) :  average time estimation 4648.5 ms
{'<CAPTION>': 'A green car parked in front of a yellow building.'}

ailia SDK (CPU)

 INFO florence2.py (517) :  ailia processing estimation time 3362 ms
 INFO florence2.py (517) :  ailia processing estimation time 2877 ms
 INFO florence2.py (517) :  ailia processing estimation time 2840 ms
 INFO florence2.py (517) :  ailia processing estimation time 2857 ms
 INFO florence2.py (517) :  ailia processing estimation time 3001 ms
 INFO florence2.py (521) :  average time estimation 2893.75 ms
{'<CAPTION>': 'A green car parked in front of a yellow building.'}

torch (CPU)

{'<CAPTION>': 'A green car parked in front of a yellow building.'}
    ailia processing estimation time 1861 ms
{'<CAPTION>': 'A green car parked in front of a yellow building.'}
    ailia processing estimation time 1884 ms
{'<CAPTION>': 'A green car parked in front of a yellow building.'}
    ailia processing estimation time 1823 ms
{'<CAPTION>': 'A green car parked in front of a yellow building.'}
    ailia processing estimation time 1895 ms
{'<CAPTION>': 'A green car parked in front of a yellow building.'}
    ailia processing estimation time 1817 ms
kyakuno commented 1 month ago

TODO : ailia_tokenizerの更新

kyakuno commented 1 month ago

モデルアップロード済み。