Why is detector head run on CPU host instead of edge TPU?

guichristmann / edge-tpu-tiny-yolo

Run Tiny YOLO-v3 on Google's Edge TPU USB Accelerator.

MIT License

102 stars 31 forks source link

Why is detector head run on CPU host instead of edge TPU? #16

Open songshan0321 opened 3 years ago

songshan0321 commented 3 years ago

Based on my understanding, the detector head in this repo (featuresToBoxes()) is run on CPU host. I wonder why it is implemented in such way because it is not a common practice due to longer inference time. Is it because edge TPU does not support all operations of detector head?