search

Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch
Apache License 2.0
498 stars 128 forks source link

ONNX Export #95

Closed kanybekasanbekov closed 1 year ago

kanybekasanbekov commented 1 year ago

I have been struggling to install vedadet on my local computer. I have encountered errors after the command pip install -v -e ., the error was about the CUDA. I spent an enormous amount of time trying to fix it, still didn't succeed. However, I was able to install, run and export to ONNX on Colab. Here is my simple code that I want to share with others: https://gist.github.com/kanybekasanbekov/9b5c9293e26905255b29c421e881e129

I hope it will be useful to someone.

TranThanh96 commented 7 months ago

@kanybekasanbekov I still can not build on colab, could you upload and share your onnx file ? many thanks

kanybekasanbekov commented 7 months ago

Hi, I've uploaded the ONNX file here

TranThanh96 commented 7 months ago

hi, thank you @kanybekasanbekov . Can you share me your detect face code with that model onnx.