computer-vision - 使用 YOLO 版本 2 进行测试

Question

运行 yolo v2 时出现以下错误。Yolo 和小 yolo 工作正常。任何线索将不胜感激

./darknet detector test ./cfg/voc.data ./cfg/yolo-voc.cfg ./yolo-     voc.weights data/dog.jpg
layer filters size input output
0 conv 32 3 x 3 / 1 416 x 416 x 3 -> 416 x 416 x 32
1 max 2 x 2 / 2 416 x 416 x 32 -> 208 x 208 x 32
2 conv 64 3 x 3 / 1 208 x 208 x 32 -> 208 x 208 x 64
3 max 2 x 2 / 2 208 x 208 x 64 -> 104 x 104 x 64
4 conv 128 3 x 3 / 1 104 x 104 x 64 -> 104 x 104 x 128
5 conv 64 1 x 1 / 1 104 x 104 x 128 -> 104 x 104 x 64
6 conv 128 3 x 3 / 1 104 x 104 x 64 -> 104 x 104 x 128
7 max 2 x 2 / 2 104 x 104 x 128 -> 52 x 52 x 128
8 conv 256 3 x 3 / 1 52 x 52 x 128 -> 52 x 52 x 256
9 conv 128 1 x 1 / 1 52 x 52 x 256 -> 52 x 52 x 128
10 conv 256 3 x 3 / 1 52 x 52 x 128 -> 52 x 52 x 256
11 max 2 x 2 / 2 52 x 52 x 256 -> 26 x 26 x 256
12 conv 512 3 x 3 / 1 26 x 26 x 256 -> 26 x 26 x 512
13 conv 256 1 x 1 / 1 26 x 26 x 512 -> 26 x 26 x 256
14 conv 512 3 x 3 / 1 26 x 26 x 256 -> 26 x 26 x 512
15 conv 256 1 x 1 / 1 26 x 26 x 512 -> 26 x 26 x 256
16 conv 512 3 x 3 / 1 26 x 26 x 256 -> 26 x 26 x 512
17 max 2 x 2 / 2 26 x 26 x 512 -> 13 x 13 x 512
18 conv 1024 3 x 3 / 1 13 x 13 x 512 -> 13 x 13 x1024
19 conv 512 1 x 1 / 1 13 x 13 x1024 -> 13 x 13 x 512
20 conv 1024 3 x 3 / 1 13 x 13 x 512 -> 13 x 13 x1024
21 conv 512 1 x 1 / 1 13 x 13 x1024 -> 13 x 13 x 512
22 conv 1024 3 x 3 / 1 13 x 13 x 512 -> 13 x 13 x1024
23 conv 1024 3 x 3 / 1 13 x 13 x1024 -> 13 x 13 x1024
24 Killed

Answer 1

此错误有时是由于内存问题而出现的。尝试释放系统 RAM 中的一些空间。或者您可以尝试更改配置文件 ./cfg/yolo-voc.cfg

更改以下参数：

批量大小：1

细分：1

这减少了处理大批量大小的开销。稍后您可以尝试缓慢增加它以进行培训。

否则，可能是 Makefile 中 ARCH 参数的问题。它应该根据您使用的 GPU 进行设置。

Answer 2

如果您在项目文件夹中，那么您的命令将是

./darknet detector test cfg/voc.data cfg/yolo-voc.cfg yolo-voc.weights data/dog.jpg

我希望这会有所帮助...检查您的.weights文件名

Answer 3

在您的配置文件中更改批量大小和细分，使它们等于 1 , 1 我猜这是内存问题

Answer 4

试试 ./darknet detect cfg/yolo-voc.cfg voc.weights data/dog.jpg。如果您仍然收到错误，您应该检查您的 .weights 文件。