* Fixed ngrapth sample logical errors
- added batch size, because sample accepts several images
- set label file as optional
}
/** Setting batch size using image count **/
- size_t batchSize = 1;
+ network.setBatchSize(imagesData.size());
+ size_t batchSize = network.getBatchSize();
slog::info << "Batch size is " << std::to_string(batchSize) << slog::endl;
// --------------------------- Prepare output blobs -----------------------------------------------------
labels.push_back(strLine);
}
inputFile.close();
- } else {
- throw std::logic_error("Cannot read label file");
}
ClassificationResult classificationResult(outputBlob, images, batchSize, FLAGS_nt, labels);