pytorch/vision

Improve error message when number of anchors mismatch

Aperta

#1539 aperta il 30 ott 2019

 (0 commenti) (0 reazioni) (0 assegnatari)Python (6858 fork)batch import
enhancementhelp wantedmodule: modelstopic: object detection

Metriche repository

Star
 (15.050 stelle)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

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.

Guida contributor