walktree / libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm
450 stars 131 forks source link

yolov3 spp 使用时推理有问题呢? #44

Open liushuan opened 4 years ago

pangxiaowen commented 4 years ago

您好,我也遇到相同的问题,是否是推理输出为nan呢?

liushuan commented 4 years ago

我已经解决了这个问题 通过修改 cfg文件  把 spp的最后一个  [route] layers=-1,-3,-5,-6 修改为:   [route] layers=-1,-3

[route] layers=-1,-6

[route] layers=-1,-8

End SPP

------------------ 原始邮件 ------------------ 发件人: "pangxiaowen"<notifications@github.com>; 发送时间: 2020年5月26日(星期二) 中午11:47 收件人: "walktree/libtorch-yolov3"<libtorch-yolov3@noreply.github.com>; 抄送: "liushuan"<547691062@qq.com>; "Author"<author@noreply.github.com>; 主题: Re: [walktree/libtorch-yolov3] yolov3 spp 使用时推理有问题呢? (#44)

您好,我也遇到相同的问题,是否是推理输出为nan呢?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

pangxiaowen commented 4 years ago

十分感谢

pangxiaowen commented 4 years ago

麻烦再问一下,为什么要这样修改呢?

liushuan commented 4 years ago

或者时修改 Darknet.cpp的源码,因为源码里面 只给了 1个和两个输入的情况。

------------------ 原始邮件 ------------------ 发件人: "pangxiaowen"<notifications@github.com>; 发送时间: 2020年5月26日(星期二) 中午12:32 收件人: "walktree/libtorch-yolov3"<libtorch-yolov3@noreply.github.com>; 抄送: "liushuan"<547691062@qq.com>; "Author"<author@noreply.github.com>; 主题: Re: [walktree/libtorch-yolov3] yolov3 spp 使用时推理有问题呢? (#44)

麻烦再问一下,为什么要这样修改呢?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.