search

zhengye1995 / DCIC19-Rebar-detection-Rank2

DCIC19 数字中国峰会钢筋智能识别 第二名 代码
18 stars 10 forks source link

nms_cuda.c not found #1

Closed monkeyshichi closed 4 years ago

monkeyshichi commented 4 years ago

when Compiling nms kernels by nvcc, /DCIC19-Rebar-detection-Rank2-master/lib/model/nms/src/nms_cuda.c: No such file or directory

monkeyshichi commented 4 years ago

请大神赐教！

zhengye1995 commented 4 years ago

Sorry, I missed this file and I upload it now.

monkeyshichi commented 4 years ago

Thanks, when i run program,it go wrong. the message is : cuda error:out of memory ,i only have 4G gpu memory ,do you have method to solve this problem ?For example, change some parameters.

zhengye1995 commented 4 years ago

Because this code is based on detectron, wich is a old version for detectron, now detectron has a new version for detectron2, and you can try it. In detectron1, it needs huge GPU memory for training and testing. If you just has a GPU with 4GB memory, I suggest you try to decrease the image size of input and use a light backbone like resnet50, but maybe it still not enough. Finally, I suggest you try detectron2 or mmdetection and other new object detection tools, detectron is so old now.

monkeyshichi commented 4 years ago

i trained with one image per batch .i will try detectron2 or mmdetection.Thanks a lot.