Open fengenxiang1123 opened 2 years ago
It seems the cuda codes of tree filter has not been compiled successfully. It could be using different cuda version for pytorch and nvcc. You can recompile the code by runing "python3 setup.py build develop --user" and check if it has similar issues.
When I execute "from cvpods.layers import TreeFilterV2", there is a problem. The detailed error is as follows: Traceback (most recent call last): File "", line 1, in
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/init.py", line 3, in
from .utils import setup_environment
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/init.py", line 27, in
from .visualizer import ColorMode, VideoVisualizer, VisImage, Visualizer, colormap, random_color
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/visualizer/init.py", line 5, in
from .video_visualizer import *
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/visualizer/video_visualizer.py", line 6, in
from .visualizer import ColorMode, Visualizer, _create_text_labels, _PanopticPrediction
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/utils/visualizer/visualizer.py", line 17, in
from cvpods.structures import BitMasks, Boxes, BoxMode, Keypoints, PolygonMasks, RotatedBoxes
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/structures/init.py", line 2, in
from .boxes import Boxes, BoxMode, pairwise_ioa, pairwise_iou
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/structures/boxes.py", line 11, in
from cvpods.layers import cat
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/layers/init.py", line 4, in
from .deform_conv import DeformConv, ModulatedDeformConv
File "/data/bbbbb/LearnableTreeFilterV2v/cvpods/layers/deform_conv.py", line 11, in
from cvpods import _C
ImportError: /data/bbbbb/LearnableTreeFilterV2v/cvpods/_C.cpython-39-x86_64-linux-gnu.so: undefined symbol: _Z39tree_filter_refine_backward_edge_weightRKN2at6TensorES2_S2_S2_S2_S2_S2_S2S2