Megvii-BaseDetection / DeFCN

End-to-End Object Detection with Fully Convolutional Network
Apache License 2.0
495 stars 36 forks source link

运行时报错 #7

Closed ChengJianjia closed 3 years ago

ChengJianjia commented 3 years ago

ImportError: cannot import name 'PSRoIAlign'

我的环境: pytorch1.3.2,torchvision0.4.2 请问是否是环境造成的错误? 请问源代码使用的具体环境是什么呢?

完整错误:Traceback (most recent call last): File "/home/cjj/cvpods/tools/train_net.py", line 28, in from cvpods.engine import DefaultTrainer, default_argument_parser, default_setup, hooks, launch File "/home/cjj/cvpods/cvpods/engine/init.py", line 3, in from .hooks import * File "/home/cjj/cvpods/cvpods/engine/hooks.py", line 16, in from cvpods.modeling.nn_utils.precise_bn import get_bn_modules, update_bn_stats File "/home/cjj/cvpods/cvpods/modeling/init.py", line 8, in from .meta_arch import GeneralizedRCNN, PanopticFPN, ProposalNetwork, RetinaNet, SemanticSegmentor File "/home/cjj/cvpods/cvpods/modeling/meta_arch/init.py", line 15, in from .pointrend import CoarseMaskHead, PointRendROIHeads, PointRendSemSegHead, StandardPointHead File "/home/cjj/cvpods/cvpods/modeling/meta_arch/pointrend.py", line 15, in from cvpods.modeling.roi_heads import StandardROIHeads, select_foreground_proposals File "/home/cjj/cvpods/cvpods/modeling/roi_heads/init.py", line 2, in from .roi_heads import Res5ROIHeads, ROIHeads, StandardROIHeads, select_foreground_proposals File "/home/cjj/cvpods/cvpods/modeling/roi_heads/roi_heads.py", line 17, in from ..poolers import ROIPooler File "/home/cjj/cvpods/cvpods/modeling/poolers.py", line 8, in from torchvision.ops import PSRoIAlign, PSRoIPool, RoIPool ImportError: cannot import name 'PSRoIAlign'

wjfwzzc commented 3 years ago

看起来是没有build,请参照cvpods安装说明安装。