dinov2 icon indicating copy to clipboard operation
dinov2 copied to clipboard

[draft]Add onnx export script

Open RRoundTable opened this issue 1 year ago • 5 comments

Objective

Export dinov2 to onnx for serving.

Changed

  • Fix hubconf to get Attention to export onnx
  • Fix minor part in model
  • Add onnx export script

Run

pip install -r requirements.txt
python script/convert-to-onnx.py

Huggingface

Check huggingface model

Related Issue

https://github.com/facebookresearch/dinov2/issues/19

RRoundTable avatar Jun 25 '23 03:06 RRoundTable