diff options
author | Tea <tea.desouza@gmail.com> | 2015-11-17 17:05:56 +0800 |
---|---|---|
committer | Tea <tea.desouza@gmail.com> | 2015-11-26 09:54:12 +0800 |
commit | b72b0318e2802785c17be1fe8ed1b6899961df19 (patch) | |
tree | efa7a1a24c420de673a7ebee343b3b1f68079907 /examples | |
parent | 8e8d97d6206cac99eae3c16baaa2275a14e64ca7 (diff) | |
download | caffeonacl-b72b0318e2802785c17be1fe8ed1b6899961df19.tar.gz caffeonacl-b72b0318e2802785c17be1fe8ed1b6899961df19.tar.bz2 caffeonacl-b72b0318e2802785c17be1fe8ed1b6899961df19.zip |
replace snprintf with a C++98 equivalent
Diffstat (limited to 'examples')
-rw-r--r-- | examples/cifar10/convert_cifar_data.cpp | 13 | ||||
-rw-r--r-- | examples/mnist/convert_mnist_data.cpp | 12 | ||||
-rw-r--r-- | examples/siamese/convert_mnist_siamese_data.cpp | 7 |
3 files changed, 14 insertions, 18 deletions
diff --git a/examples/cifar10/convert_cifar_data.cpp b/examples/cifar10/convert_cifar_data.cpp index f4c42e4d..e1b89f42 100644 --- a/examples/cifar10/convert_cifar_data.cpp +++ b/examples/cifar10/convert_cifar_data.cpp @@ -16,6 +16,7 @@ #include "caffe/proto/caffe.pb.h" #include "caffe/util/db.hpp" +#include "caffe/util/format.hpp" using caffe::Datum; using boost::scoped_ptr; @@ -52,19 +53,18 @@ void convert_dataset(const string& input_folder, const string& output_folder, for (int fileid = 0; fileid < kCIFARTrainBatches; ++fileid) { // Open files LOG(INFO) << "Training Batch " << fileid + 1; - snprintf(str_buffer, kCIFARImageNBytes, "/data_batch_%d.bin", fileid + 1); - std::ifstream data_file((input_folder + str_buffer).c_str(), + string batchFileName = input_folder + "/data_batch_" + + caffe::format_int(fileid+1) + ".bin"; + std::ifstream data_file(batchFileName.c_str(), std::ios::in | std::ios::binary); CHECK(data_file) << "Unable to open train file #" << fileid + 1; for (int itemid = 0; itemid < kCIFARBatchSize; ++itemid) { read_image(&data_file, &label, str_buffer); datum.set_label(label); datum.set_data(str_buffer, kCIFARImageNBytes); - int length = snprintf(str_buffer, kCIFARImageNBytes, "%05d", - fileid * kCIFARBatchSize + itemid); string out; CHECK(datum.SerializeToString(&out)); - txn->Put(string(str_buffer, length), out); + txn->Put(caffe::format_int(fileid * kCIFARBatchSize + itemid, 5), out); } } txn->Commit(); @@ -82,10 +82,9 @@ void convert_dataset(const string& input_folder, const string& output_folder, read_image(&data_file, &label, str_buffer); datum.set_label(label); datum.set_data(str_buffer, kCIFARImageNBytes); - int length = snprintf(str_buffer, kCIFARImageNBytes, "%05d", itemid); string out; CHECK(datum.SerializeToString(&out)); - txn->Put(string(str_buffer, length), out); + txn->Put(caffe::format_int(itemid, 5), out); } txn->Commit(); test_db->Close(); diff --git a/examples/mnist/convert_mnist_data.cpp b/examples/mnist/convert_mnist_data.cpp index 8f29bafd..16d28093 100644 --- a/examples/mnist/convert_mnist_data.cpp +++ b/examples/mnist/convert_mnist_data.cpp @@ -23,6 +23,7 @@ #include <string> #include "caffe/proto/caffe.pb.h" +#include "caffe/util/format.hpp" #if defined(USE_LEVELDB) && defined(USE_LMDB) @@ -108,8 +109,6 @@ void convert_dataset(const char* image_filename, const char* label_filename, char label; char* pixels = new char[rows * cols]; int count = 0; - const int kMaxKeyLength = 10; - char key_cstr[kMaxKeyLength]; string value; Datum datum; @@ -123,18 +122,17 @@ void convert_dataset(const char* image_filename, const char* label_filename, label_file.read(&label, 1); datum.set_data(pixels, rows*cols); datum.set_label(label); - snprintf(key_cstr, kMaxKeyLength, "%08d", item_id); + string key_str = caffe::format_int(item_id, 8); datum.SerializeToString(&value); - string keystr(key_cstr); // Put in db if (db_backend == "leveldb") { // leveldb - batch->Put(keystr, value); + batch->Put(key_str, value); } else if (db_backend == "lmdb") { // lmdb mdb_data.mv_size = value.size(); mdb_data.mv_data = reinterpret_cast<void*>(&value[0]); - mdb_key.mv_size = keystr.size(); - mdb_key.mv_data = reinterpret_cast<void*>(&keystr[0]); + mdb_key.mv_size = key_str.size(); + mdb_key.mv_data = reinterpret_cast<void*>(&key_str[0]); CHECK_EQ(mdb_put(mdb_txn, mdb_dbi, &mdb_key, &mdb_data, 0), MDB_SUCCESS) << "mdb_put failed"; } else { diff --git a/examples/siamese/convert_mnist_siamese_data.cpp b/examples/siamese/convert_mnist_siamese_data.cpp index ad08036f..928b3fbf 100644 --- a/examples/siamese/convert_mnist_siamese_data.cpp +++ b/examples/siamese/convert_mnist_siamese_data.cpp @@ -13,6 +13,7 @@ #include "stdint.h" #include "caffe/proto/caffe.pb.h" +#include "caffe/util/format.hpp" #include "caffe/util/math_functions.hpp" #ifdef USE_LEVELDB @@ -75,8 +76,6 @@ void convert_dataset(const char* image_filename, const char* label_filename, char label_i; char label_j; char* pixels = new char[2 * rows * cols]; - const int kMaxKeyLength = 10; - char key[kMaxKeyLength]; std::string value; caffe::Datum datum; @@ -99,8 +98,8 @@ void convert_dataset(const char* image_filename, const char* label_filename, datum.set_label(0); } datum.SerializeToString(&value); - snprintf(key, kMaxKeyLength, "%08d", itemid); - db->Put(leveldb::WriteOptions(), std::string(key), value); + std::string key_str = caffe::format_int(itemid, 8); + db->Put(leveldb::WriteOptions(), key_str, value); } delete db; |