nvdla / sw

NVDLA SW
Other
489 stars 193 forks source link

Dose NVDLA support MobileNet? #57

Open jimchen2018 opened 6 years ago

jimchen2018 commented 6 years ago

Dear Friends, When I use nvdla_compiler to translate mobile network, it reports error.

// SSD_MobileNet
jimchen@workstation:~/nvdla/trans$ ./nvdla_compiler --prototxt MobileNetSSD_deploy.prototxt --caffemodel MobileNetSSD_deploy.caffemodel 
creating new wisdom context...
opening wisdom context...
parsing caffe network...
[libprotobuf ERROR google/protobuf/text_format.cc:274] Error parsing text-format ditcaffe.NetParameter: 1177:17: Message type "ditcaffe.LayerParameter" has no field named "permute_param".
Caffe Parser: could not parse text file
libnvdla<1> Could not parse deploy file
(DLA) Error 0x00000004: Unable to parse caffemodel: "MobileNetSSD_deploy.prototxt" (in ParseTest.cpp, function parseCaffeNetwork(), line 91)
(DLA) Error 0x00000004: (propagating from ParseTest.cpp, function parseAndCompile(), line 164)
(DLA) Error 0x00000004: (propagating from main.cpp, function launchTest(), line 87)

// MobileNet
jimchen@workstation:~/nvdla/trans$ ./nvdla_compiler --prototxt mobilenet_deploy.prototxt --caffemodel mobilenet.caffemodel 
creating new wisdom context...
opening wisdom context...
parsing caffe network...
libnvdla<1> Parameter check failed in addPooling, condition: windowSize.h > 0
libnvdla<1> error: parsing layer type Pooling index 108
(DLA) Error 0x00000004: Unable to parse caffemodel: "mobilenet_deploy.prototxt" (in ParseTest.cpp, function parseCaffeNetwork(), line 91)
(DLA) Error 0x00000004: (propagating from ParseTest.cpp, function parseAndCompile(), line 164)
(DLA) Error 0x00000004: (propagating from main.cpp, function launchTest(), line 87)

Dose NVDLA support MobileNet and SSD?

jimchen2018 commented 6 years ago

MobileNetSSD_deploy.zip mobilenet_v2_deploy.zip mobilenet_deploy.zip

Add prototxt files.

prasshantg commented 6 years ago

@jimchen2018 MobileNet is not yet supported, will update here once we have added support for it

jimchen2018 commented 6 years ago

@prasshantg When ssd mobilenet could be supported?Thanks.