search

for-aiur / yolo_cpp

C++ed version of Yolo
Other
69 stars 37 forks source link

Why is final logistic_activate operations run on CPU host instead of GPU? #11

Open songshan0321 opened 3 years ago

songshan0321 commented 3 years ago

This get_region_box() function includes several logistic_activate operations, I wonder if there is any reason why they are not included in the model inference on GPU? Theoretically, it will be faster to offload it onto GPU instead of CPU host.