大家！ 我曾尝试根据https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects在Google实验室上使用YOLOv3。
然后，我创建自己的数据集。我做到了 ！./ darknet检测器火车数据/obj.data yolov3-obj.cfg darknet53.conv.74- dont_show -map
而且，我遇到了这样的错误
90: 1050.771118, 1071.344727 avg loss, 0.000000 rate, 23.553140 seconds,
5760 images
Resizing
576 x 576
CUDA Error: out of memory
darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.
请给我解决方法！
答案 0 :(得分：0)
看起来您的图像尺寸较大。尝试修改cfg文件，减小宽度和高度。
例如宽度= 416，高度= 416
如果仍然没有足够的内存，请试一下批处理大小和细分编号。
示例：
批次= 64 subdivision = 64
或
批次= 32 subdivision = 32
顺便说一句，您在使用GPU吗？如果是，则可以尝试运行Nvidia Nsight Monitor，然后右键单击任务栏图标（屏幕右下方），然后选择“选项”。在“常规”>“ Microsoft显示驱动程序”>“ WDDM TDR延迟”下，将其更改为30或32。