ignacio-rocco / detectorch

Detectorch - detectron for PyTorch
Other
558 stars 72 forks source link

How to build ROIAlign for pytorch 0.4 #16

Open MathewSam opened 6 years ago

MathewSam commented 6 years ago

Do we have to do the same operations for both pytorch 0.3.1 and pytorch 0.4 to build ROIAlign? The demo file keeps restarting every time I pass a sample image to the model and this is the only thing I am doubtful about. Thanks