search

digitalbrain79 / darknet-nnpack

Darknet with NNPACK
Other
305 stars 75 forks source link

Out of mem when classifier image #42

Closed smallworld-network-wupeng closed 4 years ago

smallworld-network-wupeng commented 4 years ago 
../darknet-nnpack/darknet classifier predict cifar.data tiny.cfg tiny_final.weights

layer     filters    size              input                output
    0 conv     16  3 x 3 / 1   128 x 128 x   3   ->   128 x 128 x  16  0.014 BFLOPs
    1 max          2 x 2 / 2   128 x 128 x  16   ->    64 x  64 x  16
    2 conv     32  3 x 3 / 1    64 x  64 x  16   ->    64 x  64 x  32  0.038 BFLOPs
    3 max          2 x 2 / 2    64 x  64 x  32   ->    32 x  32 x  32
    4 conv     16  1 x 1 / 1    32 x  32 x  32   ->    32 x  32 x  16  0.001 BFLOPs
    5 conv    128  3 x 3 / 1    32 x  32 x  16   ->    32 x  32 x 128  0.038 BFLOPs
    6 conv     16  1 x 1 / 1    32 x  32 x 128   ->    32 x  32 x  16  0.004 BFLOPs
    7 conv    128  3 x 3 / 1    32 x  32 x  16   ->    32 x  32 x 128  0.038 BFLOPs
    8 max          2 x 2 / 2    32 x  32 x 128   ->    16 x  16 x 128
    9 conv     32  1 x 1 / 1    16 x  16 x 128   ->    16 x  16 x  32  0.002 BFLOPs
   10 conv    256  3 x 3 / 1    16 x  16 x  32   ->    16 x  16 x 256  0.038 BFLOPs
   11 conv     32  1 x 1 / 1    16 x  16 x 256   ->    16 x  16 x  32  0.004 BFLOPs
   12 conv    256  3 x 3 / 1    16 x  16 x  32   ->    16 x  16 x 256  0.038 BFLOPs
   13 max          2 x 2 / 2    16 x  16 x 256   ->     8 x   8 x 256
   14 conv     64  1 x 1 / 1     8 x   8 x 256   ->     8 x   8 x  64  0.002 BFLOPs
   15 conv    512  3 x 3 / 1     8 x   8 x  64   ->     8 x   8 x 512  0.038 BFLOPs
   16 conv     64  1 x 1 / 1     8 x   8 x 512   ->     8 x   8 x  64  0.004 BFLOPs
   17 conv    512  3 x 3 / 1     8 x   8 x  64   ->     8 x   8 x 512  0.038 BFLOPs
   18 conv    128  1 x 1 / 1     8 x   8 x 512   ->     8 x   8 x 128  0.008 BFLOPs
   19 conv      9  1 x 1 / 1     8 x   8 x 128   ->     8 x   8 x   9  0.000 BFLOPs
   20 avg                        8 x   8 x   9   ->     9
   21 softmax                                           9
Loading weights from tiny_final.weights...Done!
Enter Image Path: test.jpg
Cannot load image "test.jpg"
STB Reason:  outofmem

When I classifier image , there is out of mem .

smallworld-network-wupeng commented 4 years ago

The new version has fixed this problem