fixed compilation on 3090

CMakeLists.txt

@@ -4,7 +4,7 @@ project(BundleTrack)
 
 set(CMAKE_BUILD_TYPE Release)
 add_compile_options(-std=c++14 -fpermissive -fPIC -g)
-set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} -Xcompiler;-fPIC;-gencode arch=compute_30,code=sm_30;-gencode arch=compute_61,code=sm_61;-gencode arch=compute_75,code=sm_75;-O3;-std=c++11;-use_fast_math;--default-stream per-thread)
+set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} -Xcompiler;-fPIC;-gencode arch=compute_61,code=sm_61;-gencode arch=compute_75,code=sm_75;-O3;-std=c++11;-use_fast_math;--default-stream per-thread)
 
 if (${CMAKE_BUILD_TYPE} STREQUAL "Release")
   set(COMPILE_OPTIONS -fopenmp -march=native -msse -msse2 -msse3 -msse4 -msse4.2 -mavx2 -ftree-vectorize)

docker/run_container.sh

@@ -5,4 +5,4 @@ echo "BUNDLETRACK_DIR $BUNDLETRACK_DIR"
 echo "NOCS_DIR $NOCS_DIR"
 echo "YCBINEOAT_DIR $YCBINEOAT_DIR"
 
-docker run --gpus all -it --network=host --name bundletrack  -m  16000m --cap-add=SYS_PTRACE --security-opt seccomp=unconfined  -v $BUNDLETRACK_DIR:$BUNDLETRACK_DIR:rw -v $NOCS_DIR:$NOCS_DIR -v $YCBINEOAT_DIR:$YCBINEOAT_DIR -v /tmp:/tmp  --ipc=host -e DISPLAY=${DISPLAY} -e GIT_INDEX_FILE wenbowen123/bundletrack:latest bash
+docker run --gpus all -it --network=host --name bundletrack  -m  16000m --cap-add=SYS_PTRACE --security-opt seccomp=unconfined  -v $BUNDLETRACK_DIR:$BUNDLETRACK_DIR:rw -v $NOCS_DIR:$NOCS_DIR -v $YCBINEOAT_DIR:$YCBINEOAT_DIR -v /tmp:/tmp  --ipc=host -e DISPLAY=${DISPLAY} -e GIT_INDEX_FILE wenbowen123/bundletrack:3090 bash

src/Utils.cpp

@@ -33,11 +33,6 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
 
 #include "Utils.h"
-#include <vtkPolyLine.h>
-#include <vtkImageReader2Factory.h>
-#include <vtkImageReader2.h>
-#include <vtkImageData.h>
-#include <vtkImageFlip.h>
 
 
 namespace Utils

src/Utils.h

@@ -77,9 +77,6 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #include <opencv2/rgbd.hpp>
 #include "opencv2/calib3d/calib3d.hpp"
 
-// For Visualization
-// #include <pcl/visualization/pcl_visualizer.h>
-// #include <pcl/visualization/cloud_viewer.h>
 
 #include <pcl/point_types.h>
 #include <pcl/point_cloud.h>