different implemention of gpu and cpu version of nms_layer

CMU-Perceptual-Computing-Lab / caffe_rtpose

Realtime C++ code for multi-person pose estimation

Other

357 stars 207 forks source link

different implemention of gpu and cpu version of nms_layer #4

Closed KongCDY closed 7 years ago

KongCDY commented 7 years ago

i found that the implemention of cpu and gpu forward function of nms_layer is different

tsimk commented 7 years ago

Yes, we don't really use the CPU implementations so they are in need of debugging. There's currently a bug where it doesn't check that peakCount doesn't exceed max_peaks.

We'd be interested in knowing if you end up using the CPU version.

KongCDY commented 7 years ago

I just see how nms_layer implement, so I check the cpu implementation, the implementation is different from the result output of net, so I check the gpu implementation.