Try this snippet

  
      gist.github.com
  
  
    https://gist.github.com/ruotianluo/043fae22e9f8fd1b36b82189f2356937 94
test_roialign.py
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable

import numpy as np
This file has been truncated. show original