Merge pull request #888 from ronghanghu/matcaffe-add-check

add necessary input checks for matcaffe
This commit is contained in:
Sergio Guadarrama 2014-08-09 02:29:05 -07:00
Родитель c5a1aa51a5 c8b6457de5
Коммит c13fded3ac
1 изменённых файлов: 4 добавлений и 0 удалений

Просмотреть файл

@ -50,6 +50,10 @@ static mxArray* do_forward(const mxArray* const bottom) {
input_blobs.size());
for (unsigned int i = 0; i < input_blobs.size(); ++i) {
const mxArray* const elem = mxGetCell(bottom, i);
CHECK(mxIsSingle(elem))
<< "MatCaffe require single-precision float point data";
CHECK_EQ(mxGetNumberOfElements(elem), input_blobs[i]->count())
<< "MatCaffe input size does not match the input size of the network";
const float* const data_ptr =
reinterpret_cast<const float* const>(mxGetPr(elem));
switch (Caffe::mode()) {