jwyang/faster-rcnn.pytorch

NMS_GPU different results from NMS_CPU

Open

#220 创建于 2018年6月29日

在 GitHub 查看
 (19 评论) (0 反应) (0 负责人)Python (2,342 fork)batch import
help wanted

仓库指标

Star
 (7,454 star)
PR 合并指标
 (30 天内没有已合并 PR)

描述

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

贡献者指南