eaphan / upidet Goto Github PK

python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml

Traceback (most recent call last):
File "/home/karker/anaconda3/envs/bp/lib/python3.7/runpy.py", line 183, in _run_module_as_main
mod_name, mod_spec, code = _get_module_details(mod_name, _Error)
File "/home/karker/anaconda3/envs/bp/lib/python3.7/runpy.py", line 109, in _get_module_details
import(pkg_name)
File "/home/karker/PROJ/BiProDet-main/OpenPCDet/pcdet/datasets/init.py", line 15, in
from .argo2.argo2_dataset import Argo2Dataset
File "/home/karker/PROJ/BiProDet-main/OpenPCDet/pcdet/datasets/argo2/argo2_dataset.py", line 7, in
from av2.utils.io import read_feather
ModuleNotFoundError: No module named 'av2'
这个av2是什么，我没有找到这个包

python3 setup.py develop --user
/usr/lib/python3/dist-packages/setuptools/dist.py:473: UserWarning: Normalizing '0.5.2+0000000' to '0.5.2+0'
warnings.warn(
running develop
running egg_info
writing pcdet.egg-info/PKG-INFO
writing dependency_links to pcdet.egg-info/dependency_links.txt
writing requirements to pcdet.egg-info/requires.txt
writing top-level names to pcdet.egg-info/top_level.txt
reading manifest file 'pcdet.egg-info/SOURCES.txt'
writing manifest file 'pcdet.egg-info/SOURCES.txt'
running build_ext
building 'pcdet.ops.iou3d_nms.iou3d_nms_cuda' extension
Emitting ninja build file /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/build.ninja...
Compiling objects...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
x86_64-linux-gnu-g++ -pthread -shared -Wl,-O1 -Wl,-Bsymbolic-functions -Wl,-Bsymbolic-functions -Wl,-z,relro -g -fwrapv -O2 -Wl,-Bsymbolic-functions -Wl,-z,relro -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/iou3d_nms/src/iou3d_cpu.o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/iou3d_nms/src/iou3d_nms_api.o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/iou3d_nms/src/iou3d_nms.o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/iou3d_nms/src/iou3d_nms_kernel.o -L/home/amine/.local/lib/python3.8/site-packages/torch/lib -L/usr/local/cuda-11.6/lib64 -lc10 -ltorch -ltorch_cpu -ltorch_python -lcudart -lc10_cuda -ltorch_cuda_cu -ltorch_cuda_cpp -o build/lib.linux-x86_64-3.8/pcdet/ops/iou3d_nms/iou3d_nms_cuda.cpython-38-x86_64-linux-gnu.so
building 'pcdet.ops.roiaware_pool3d.roiaware_pool3d_cuda' extension
Emitting ninja build file /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/build.ninja...
Compiling objects...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
x86_64-linux-gnu-g++ -pthread -shared -Wl,-O1 -Wl,-Bsymbolic-functions -Wl,-Bsymbolic-functions -Wl,-z,relro -g -fwrapv -O2 -Wl,-Bsymbolic-functions -Wl,-z,relro -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/roiaware_pool3d/src/roiaware_pool3d.o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/roiaware_pool3d/src/roiaware_pool3d_kernel.o -L/home/amine/.local/lib/python3.8/site-packages/torch/lib -L/usr/local/cuda-11.6/lib64 -lc10 -ltorch -ltorch_cpu -ltorch_python -lcudart -lc10_cuda -ltorch_cuda_cu -ltorch_cuda_cpp -o build/lib.linux-x86_64-3.8/pcdet/ops/roiaware_pool3d/roiaware_pool3d_cuda.cpython-38-x86_64-linux-gnu.so
building 'pcdet.ops.roipoint_pool3d.roipoint_pool3d_cuda' extension
Emitting ninja build file /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/build.ninja...
Compiling objects...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
x86_64-linux-gnu-g++ -pthread -shared -Wl,-O1 -Wl,-Bsymbolic-functions -Wl,-Bsymbolic-functions -Wl,-z,relro -g -fwrapv -O2 -Wl,-Bsymbolic-functions -Wl,-z,relro -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/roipoint_pool3d/src/roipoint_pool3d.o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/roipoint_pool3d/src/roipoint_pool3d_kernel.o -L/home/amine/.local/lib/python3.8/site-packages/torch/lib -L/usr/local/cuda-11.6/lib64 -lc10 -ltorch -ltorch_cpu -ltorch_python -lcudart -lc10_cuda -ltorch_cuda_cu -ltorch_cuda_cpp -o build/lib.linux-x86_64-3.8/pcdet/ops/roipoint_pool3d/roipoint_pool3d_cuda.cpython-38-x86_64-linux-gnu.so
building 'pcdet.ops.pointnet2.pointnet2_stack.pointnet2_stack_cuda' extension
Emitting ninja build file /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/build.ninja...
Compiling objects...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
[1/10] c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
FAILED: /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.o
c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
/home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/group_points.cpp:12:10: fatal error: THC/THC.h: No such file or directory
12 | #include <THC/THC.h>
| ^~~~~~~~~~~
compilation terminated.
[2/10] c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
FAILED: /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.o
c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
/home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query.cpp:3:10: fatal error: THC/THC.h: No such file or directory
3 | #include <THC/THC.h>
| ^~~~~~~~~~~
compilation terminated.
[3/10] c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
FAILED: /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.o
c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
/home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate.cpp:10:10: fatal error: THC/THC.h: No such file or directory
10 | #include <THC/THC.h>
| ^~~~~~~~~~~
compilation terminated.
[4/10] c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
FAILED: /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.o
c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
/home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query.cpp:10:10: fatal error: THC/THC.h: No such file or directory
10 | #include <THC/THC.h>
| ^~~~~~~~~~~
compilation terminated.
[5/10] c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
FAILED: /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.o
c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
/home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/ball_query_count.cpp:10:10: fatal error: THC/THC.h: No such file or directory
10 | #include <THC/THC.h>
| ^~~~~~~~~~~
compilation terminated.
[6/10] c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
FAILED: /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.o
c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
/home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool.cpp:13:10: fatal error: THC/THC.h: No such file or directory
13 | #include <THC/THC.h>
| ^~~~~~~~~~~
compilation terminated.
[7/10] c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
FAILED: /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.o
c++ -MMD -MF /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.o.d -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -O2 -Wall -g -fstack-protector-strong -Wformat -Werror=format-security -g -fwrapv -O2 -g -fstack-protector-strong -Wformat -Werror=format-security -Wdate-time -D_FORTIFY_SOURCE=2 -fPIC -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.cpp -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.o -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -std=c++14
/home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/sampling.cpp:4:10: fatal error: THC/THC.h: No such file or directory
4 | #include <THC/THC.h>
| ^~~~~~~~~~~
compilation terminated.
[8/10] /usr/local/cuda-11.6/bin/nvcc -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool_gpu.cu -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/vector_pool_gpu.o -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS -D__CUDA_NO_BFLOAT16_CONVERSIONS -D__CUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_50,code=compute_50 -gencode=arch=compute_50,code=sm_50 -std=c++14
[9/10] /usr/local/cuda-11.6/bin/nvcc -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate_gpu.cu -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/interpolate_gpu.o -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS -D__CUDA_NO_BFLOAT16_CONVERSIONS -D__CUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="gcc"' '-DPYBIND11_STDLIB="libstdcpp"' '-DPYBIND11_BUILD_ABI="cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_50,code=compute_50 -gencode=arch=compute_50,code=sm_50 -std=c++14
[10/10] /usr/local/cuda-11.6/bin/nvcc -I/home/amine/.local/lib/python3.8/site-packages/torch/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/torch/csrc/api/include -I/home/amine/.local/lib/python3.8/site-packages/torch/include/TH -I/home/amine/.local/lib/python3.8/site-packages/torch/include/THC -I/usr/local/cuda-11.6/include -I/usr/include/python3.8 -c -c /home/amine/OpenPCDet/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query_gpu.cu -o /home/amine/OpenPCDet/build/temp.linux-x86_64-3.8/pcdet/ops/pointnet2/pointnet2_stack/src/voxel_query_gpu.o -D__CUDA_NO_HALF_OPERATORS -D__CUDA_NO_HALF_CONVERSIONS -D__CUDA_NO_BFLOAT16_CONVERSIONS -D__CUDA_NO_HALF2_OPERATORS --expt-relaxed-constexpr --compiler-options ''"'"'-fPIC'"'"'' -DTORCH_API_INCLUDE_EXTENSION_H '-DPYBIND11_COMPILER_TYPE="_gcc"' '-DPYBIND11_STDLIB="_libstdcpp"' '-DPYBIND11_BUILD_ABI="_cxxabi1011"' -DTORCH_EXTENSION_NAME=pointnet2_stack_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_50,code=compute_50 -gencode=arch=compute_50,code=sm_50 -std=c++14
ninja: build stopped: subcommand failed.
Traceback (most recent call last):
File "/home/amine/.local/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1900, in _run_ninja_build
subprocess.run(
File "/usr/lib/python3.8/subprocess.py", line 516, in run
raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "setup.py", line 34, in
setup(
File "/usr/lib/python3/dist-packages/setuptools/init.py", line 144, in setup
return distutils.core.setup(**attrs)
File "/usr/lib/python3.8/distutils/core.py", line 148, in setup
dist.run_commands()
File "/usr/lib/python3.8/distutils/dist.py", line 966, in run_commands
self.run_command(cmd)
File "/usr/lib/python3.8/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/usr/lib/python3/dist-packages/setuptools/command/develop.py", line 38, in run
self.install_for_development()
File "/usr/lib/python3/dist-packages/setuptools/command/develop.py", line 140, in install_for_development
self.run_command('build_ext')
File "/usr/lib/python3.8/distutils/cmd.py", line 313, in run_command
self.distribution.run_command(command)
File "/usr/lib/python3.8/distutils/dist.py", line 985, in run_command
cmd_obj.run()
File "/usr/lib/python3/dist-packages/setuptools/command/build_ext.py", line 87, in run
_build_ext.run(self)
File "/usr/lib/python3.8/distutils/command/build_ext.py", line 340, in run
self.build_extensions()
File "/home/amine/.local/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 843, in build_extensions
build_ext.build_extensions(self)
File "/usr/lib/python3.8/distutils/command/build_ext.py", line 449, in build_extensions
self._build_extensions_serial()
File "/usr/lib/python3.8/distutils/command/build_ext.py", line 474, in _build_extensions_serial
self.build_extension(ext)
File "/usr/lib/python3/dist-packages/setuptools/command/build_ext.py", line 208, in build_extension
_build_ext.build_extension(self, ext)
File "/usr/lib/python3.8/distutils/command/build_ext.py", line 528, in build_extension
objects = self.compiler.compile(sources,
File "/home/amine/.local/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 658, in unix_wrap_ninja_compile
_write_ninja_file_and_compile_objects(
File "/home/amine/.local/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1573, in _write_ninja_file_and_compile_objects
_run_ninja_build(
File "/home/amine/.local/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1916, in _run_ninja_build
raise RuntimeError(message) from e
RuntimeError: Error compiling objects for extension

Hello, can someone explain why the kitti dataset contains different calibration files with different values ?
As far as i know the calibration is made once since the position of sensors is known.

Hi I have a dataset composed of images from a single camera, point clouds from lidar vlp-16, calibration.
I am struggling with executing the inference on it.

Thank you very much for sharing the code! Can this program run on waymo or nuscense dataset？ and if so, which parts need to be adjusted? Looking forward to your reply.

Hi, Thank you for sharing your work!

When generating *train.pkl using kitti_dataste.py, I don't think 'segmenation label' is included.

Can you give me some advise for fixing this issue?

Thank you.

Hello, I would like to ask about the paper architecture of bidirectional propagation. What is its written form or how is it named? I have searched through the code in pcdet/models/backbones_3d, pcdet/models/dense_heads, pcdet/models/detectors, and pcdet/models/model_utils, but I have not found any code that starts name with "Bi" or "propagation". Could anyone please let me know where the main architecture of this paper is written? Thank you.

I want to generate segmentation labels for nuscenes and waymo. What should I do for codes?

How to visualize UPIDet

I tested the pre-trained model under pytorch1.10.0, cuda11.3,python=3.8, and the result is different from the published result, is it because of the version of pytorch?

Dear author! Thank you very much for the code you provided. I would like to ask, I do not want to perform two-dimensional segmentation tasks (segmentation tasks are temporarily useless to me and will increase training time). How should I adjust in the code?

epochs: 0%| | 0/80 [00:00<?, ?it/s]
Traceback (most recent call last):
File "train.py", line 202, in
main()
File "train.py", line 154, in main
train_model(
File "/202121622952/UPIDet/tools/train_utils/train_utils.py", line 111, in train_model
accumulated_iter = train_one_epoch(
File "/202121622952/UPIDet/tools/train_utils/train_utils.py", line 25, in train_one_epoch
batch = next(dataloader_iter)
File "/root/anaconda3/envs/upi/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 530, in next
data = self._next_data()
File "/root/anaconda3/envs/upi/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 570, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/root/anaconda3/envs/upi/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/anaconda3/envs/upi/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/202121622952/UPIDet/tools/../pcdet/datasets/kitti/kitti_dataset.py", line 407, in getitem
input_dict['segmentation'] = annos["inmodal_seg"]
KeyError: 'inmodal_seg'

I print（annos）,but i find annos isn't include 'inmodal_seg':

'name': array(['Car', 'DontCare', 'DontCare'], dtype='<U8'), 'truncated': array([ 0., -1., -1.]), 'occluded': array([ 0., -1., -1.]), 'alpha': array([ 1.92, -10. , -10. ]), 'bbox': array([[254.08, 184.29, 360.28, 232.81],
[583.3 , 181.83, 593.3 , 191.83],
[500.5 , 177.94, 520.2 , 190.52]], dtype=float32), 'dimensions': array([[ 4.33, 1.48, 1.97],
[-1. , -1. , -1. ],
[-1. , -1. , -1. ]]), 'location': array([[ -10.3 , 1.89, 24.82],
[-1000. , -1000. , -1000. ],
[-1000. , -1000. , -1000. ]], dtype=float32), 'rotation_y': array([ 1.53, -10. , -10. ]), 'score': array([-1., -1., -1.]), 'difficulty': array([ 0, -1, -1], dtype=int32), 'index': array([ 0, -1, -1], dtype=int32), 'gt_boxes_lidar': array([[25.10887909, 10.32051182, -0.85387194, 4.33 , 1.97 ,
1.48 , -3.10079633]]), 'num_points_in_gt': array([163, -1, -1], dtype=int32)}

Hi,
Thank you for your repo. For distributed training, I was unable to locate scripts/dist_train.sh file. It would be helpful if you can share the file/link to access it.
Thanks in advance.

Hello! How can I train the whole KITTI train set consisting of 7481 frames?

Hi,
Thank you for the excellent repo. I used multi gpus to train the model from scratch and when I check the performance of the network on validation set, it is little lower compared to the pre-trained checkpoint provided.
Any additional changes/suggestions for network training would be really helpful.
Regards
Niranjan.

Thank you for shareing.
There are some problems when trying train or test using tools/cfgs/kitti_models/biprodet.yaml.

A. training:

1. segmentation data is needed. Even if 'segmentation' removed from the above config, data augmentation using segmentation is hard-coded in the code in datasets_sampler.py

B. Test:
Seems the model shared in google drive only contains weights in lidar parts, no image backbone or roi parts.

I am trying to test the model on my custom dataset (images and their corresponding vlp-16 point cloud), after that I need to train the annotate the dataset in order to train the model but i think the problem is with the .pkl files i need to find a solution. Please help me with the steps if you can and thanks.

I am trying to reduce the number of samples for training/testing purpose but I wasn't able to dot that since I have to use the provided pickle info files for segmentation.

Please provide a demo file for biprodet with 3D bounding boxes for image and point cloud.

Hi,
Thanks for your great work. Currently, I am trying to do some experiments on my customized data. Could you please tell me how to generate the semantic segmentation label in the info files? Thanks!