matlab-deep-learning / mtcnn-face-detection

Face detection and alignment using deep learning
Other
26 stars 18 forks source link

Account for new gpuArray support in selectStrongestBboxes #26

Closed justinpinkney closed 3 years ago

justinpinkney commented 3 years ago

Fixes #20.

selectStrongestBbox got gpuArray support in MATLAB R2020b. Previously the gather was required if bboxes was gpuArray and needed to be on the cpu (strangely the scores didn't matter).

In R2020b they need to be the same either both on the gpu or not, so we do a version check and adjust behaviour accordingly