jwyang/faster-rcnn.pytorch

NMS_GPU different results from NMS_CPU

Open

#220 aberto em 29 de jun. de 2018

Ver no GitHub
 (19 comments) (0 reactions) (0 assignees)Python (2.342 forks)batch import
help wanted

Métricas do repositório

Stars
 (7.454 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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

Guia do colaborador

Direção de pesquisa
Investigue a implementação GPU do NMS e compare com a implementação CPU para encontrar onde os resultados divergem. Rastreie o fluxo de dados e verifique diferenças no alinhamento de índices ou no tratamento de limiares.
Pilha de tecnologia
pythonpytorch
Domain
computer visionmachine learning
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Principalmente claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
70