detectorch
detectorch copied to clipboard
How to build ROIAlign for pytorch 0.4
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