pycharm安装pytorch_RoIAlignforPyTorch的安装,和使⽤
cro。。。
前⼏天了⼀个代码,⾥⾯要⽤PyTorch版的RoIAlign函数,⾥⾯有⼀句
p_and_resize import CropAndResizeFunction
roialign是代码压缩包⾥包含的⼀个⽂件夹,⼀直报错,然后我就绞尽脑汁的想到底咋能调⽤这个⽂件夹⾥的函数啊。
结果,谁知道,这玩意根本不⽤从⽂件夹调⽤函数,⼈家直接装在你系统上了,把⽂件夹的路径删了就⾏了,……我……这个程序他到底是怎么运⾏起来的,他真的能运⾏吗
总结下糟⼼的历程:
下⾯是RoIAlign for PyTorch的安装包pycharm安装教程和使用
github/longcw/RoIAlign.pytorch/tree/pytorch_0.4g ithub
⾸先需要gcc版本<6
然后sudo sh ./install.sh
然后sudo sh ./test.sh 应该会⽣成⼀个.so⽂件,然后如果调⽤⽂件夹写好的函数会⽤到这个⽂件。
不知道为什么,我直接sudo sh ./test.sh总是会报错⽆法导⼊tensorflow。但是我发现这个sh ./test.sh其实就是运⾏的tests⽂件夹下的test.py⽂件。所以就直接运⾏test.py,我赌它有⼀样的效果。
python test.py 还真⽣成了_crop_and_resize.so了!
但是这个时候还是⽆法读取_crop_and_resize.so,我真的要绝望了,好像永远⽆法搞定这个
但是,我发现test.py直接能⽤:
from p_and_resize import CropAndResizeFunction
⽽我的最终⽬标不就是导⼊这个糟⼼函数吗?既然能运⾏,说明这句话执⾏没问题,那说明之前的位置也能直接导⼊函数,⽽不需要打开什么辣鸡⽂件夹?!然后我把原来那个蛊惑众⽣的roialign⽂件夹⼀删,直接当成roi_align是⼀个安装在系统上的包那样导⼊它:
from p_and_resize import CropAndResizeFunction
呵呵,真的可以。
所以不需要那个.so⽂件也⾏。我真的不明⽩作者为什么要写从⽂件夹导⼊,为了这玩意我踏马快疯了。
下⾯是⽀持pytorch1.0的版本
longcw/RoIAlign.pytorch g ithub