kornia/kornia

[Feature]: Replace manual DETR box processing with Kornia BoundingBox API

Open

#3.530 aberto em 27 de jan. de 2026

Ver no GitHub
 (4 comments) (0 reactions) (1 assignee)Python (892 forks)batch import
help wantedtriage

Métricas do repositório

Stars
 (8.677 stars)
Métricas de merge de PR
 (Mesclagem média 19d 9h) (7 fundiu PRs em 30d)

Description

Title: [Refactor] Replace manual DETR box processing with Kornia BoundingBox API

Description

The forward method in the DETR post-processor currently performs manual tensor arithmetic to convert bounding box coordinates and formats:

# Manual conversion from cxcywh -> xywh
cxcy, wh = boxes[..., :2], boxes[..., 2:]
boxes_xy = torch.cat([cxcy - wh * 0.5, wh], -1)

There are TODO and NOTE comments explicitly suggesting the replacement of this manual logic with Kornia's BoundingBox API or geometry functional API.

Using raw tensor operations for geometry is error-prone and less readable.

Goal

Refactor the post-processing logic to use Kornia's geometry primitives.

Guia do colaborador

Direção de pesquisa
Familiarize se com a API BoundingBox do Kornia, especialmente a conversão de cxcywh para xywh. Em seguida, substitua as operações manuais de tensor no método forward do pós processador DETR pelas funções apropriadas do Kornia. Certifique se de que os testes existentes passem.
Pilha de tecnologia
pythonpytorch
Domain
aiapi
Tipo Issue
Refatoração
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.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Recente
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
40