How to run detection network?

ARM-software / ComputeLibrary

The Compute Library is a set of computer vision and machine learning functions optimised for both Arm CPUs and GPUs using SIMD technologies.

MIT License

2.83k stars 775 forks source link

How to run detection network? #634

Closed hiigao closed 5 years ago

hiigao commented 5 years ago

Problem description: I want to run the detection network, but only Yolo3 are provided among the examples. Can I run SSD based detection networks on the ComputeLibrary? By the way, I mainly use the GPU to run the model.

ManubARM commented 5 years ago

Hi @hiigao, It is possible, but you have to create the graph and give it the weights. In this issue: https://github.com/ARM-software/ComputeLibrary/issues/617 you can find how to extract the weights from a tensorflow pre-trained MobilenetV2 SSDlite frozen graph

Best, Manuel

ManubARM commented 5 years ago

Closing for inactivity, feel free to reopen otherwise