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_S2_S2
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.
StevenGrove
