adityaarun1 / Detectron.pytorch

A PyTorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available. Supports PyTorch 1.1+ and TorchVision 0.3.0+. No compilation necessary.
MIT License
65 stars 18 forks source link

Update model_builder.py #9

Open whytin opened 5 years ago

whytin commented 5 years ago

L279 Correct RoIAlignFunction() to ROIAlgin()

adityaarun1 commented 5 years ago

Thanks. The master branch already has this fixed [refer here]. The pytorch-1.0 branch is obsolete.