help wanted
Métriques du dépôt
- Stars
- (7 454 stars)
- Métriques de merge PR
- (Aucune PR mergée en 30 j)
Description
I compiled successfully in Ubuntu 14.04 and I use Python 3.6
import numpy as np
detections = torch.from_numpy(np.array([
(12, 84, 140, 212, 0.1),
(24, 84, 152, 212, 0.8),
(36, 84, 164, 212, 0.7),
(12, 96, 140, 224, 0.6),
(24, 96, 152, 224, 0.5),
(24, 108, 152, 236, 0.9)
]))
# print(detections)
print(nms_gpu(detections, 0.1))
print(nms_cpu(detections, 0.1))
## print results below
tensor([], dtype=torch.int32)
tensor([ 5], dtype=torch.int32)
You can see here, the CPU results is correct, whereas the GPU results is not returning anything