thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

New optimized NMS

Browse Source
This commit is contained in:
Marcos Luciano
2022-08-15 02:48:23 -03:00
parent 8ad99d3f8d
commit c8a4a49f16
24 changed files with 206 additions and 394 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
nvdsinfer_custom_impl_Yolo/Makefile
View File

@@ -69,12 +69,8 @@ all: $(TARGET_LIB)
%.o: %.cpp $(INCS) Makefile
$(CC) -c $(COMMON) -o $@ $(CFLAGS) $<
ifeq ($(CUDA_VER), 10.2)
CUB=-I/usr/local/cuda-$(CUDA_VER)/include/thrust/system/cuda/detail
endif
%.o: %.cu $(INCS) Makefile
$(NVCC) -c -o $@ $(CUB) --compiler-options '-fPIC' $<
$(NVCC) -c -o $@ --compiler-options '-fPIC' $<
$(TARGET_LIB) : $(TARGET_OBJS)
$(CC) -o $@ $(TARGET_OBJS) $(LFLAGS)