pytorch/vision

Improve error message when number of anchors mismatch

Open

#1 539 ouverte le 30 oct. 2019

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)Python (6 858 forks)batch import
enhancementhelp wantedmodule: modelstopic: object detection

Métriques du dépôt

Stars
 (15 050 stars)
Métriques de merge PR
 (Merge moyen 12j 8h) (14 PRs mergées en 30 j)

Description

See discussion in https://github.com/pytorch/vision/issues/1308#issuecomment-547964507

We should add checks in https://github.com/pytorch/vision/blob/067704986d92a3dff4ccf4a686080b3d31390402/torchvision/models/detection/rpn.py#L440 so that it's clearer to users what they need to change if they face such an error in the future.

Guide contributeur