Closed monkeyshichi closed 4 years ago
请大神赐教!
Sorry, I missed this file and I upload it now.
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.
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.
i trained with one image per batch .i will try detectron2 or mmdetection.Thanks a lot.
when Compiling nms kernels by nvcc, /DCIC19-Rebar-detection-Rank2-master/lib/model/nms/src/nms_cuda.c: No such file or directory