//
//?This?script?converts?the?CIFAR?dataset?to?the?leveldb?format?used
//?by?caffe?to?perform?classification.
//?Usage:
//????convert_cifar_data?input_folder?output_db_file
//?The?CIFAR?dataset?could?be?downloaded?at
//????http://www.cs.toronto.edu/~kriz/cifar.html

#include???//?NOLINT（readability/streams）
#include?

#include?“boost/scoped_ptr.hpp“
#include?“glog/logging.h“
#include?“google/protobuf/text_format.h“
#include?“stdint.h“

#include?“caffe/proto/caffe.pb.h“
#include?“caffe/util/db.hpp“
#include?“caffe/util/format.hpp“

using?caffe::Datum;
using?boost::scoped_ptr;
using?std::string;
namespace?db?=?caffe::db;

const?int?kCIFARSize?=?32;
const?int?kCIFARImageNBytes?=?3072;
const?int?kCIFARBatchSize?=?10000;
const?int?kCIFARTrainBatches?=?5;

void?read_image（std::ifstream*?file?int*?label?char*?buffer）?{
??char?label_char;
??file->read（&label_char?1）;
??*label?=?label_char;
??file->read（buffer?kCIFARImageNBytes）;
??return;
}

void?convert_dataset（const?string&?input_folder?const?string&?output_folder
????const?string&?db_type）?{
??scoped_ptr?train_db（db::GetDB（db_type））;
??train_db->Open（output_folder?+?“/cifar10_train_“?+?db_type?db::NEW）;
??scoped_ptr?txn（train_db->NewTransaction（））;
??//?Data?buffer
??int?label;
??char?str_buffer[kCIFARImageNBytes];
??Datum?datum;
??datum.set_channels（3）;
??datum.set_height（kCIFARSize）;
??datum.set_width（kCIFARSize）;

??LOG（INFO）?<??for?（int?fileid?=?0;?fileid?????//?Open?files
????LOG（INFO）?<????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）?<????for?（int?itemid?=?0;?itemid???????read_image（&data_file?&label?str_buffer）;
??????datum.set_label（label）;
??????datum.set_data（str_buffer?kCIFARImageNBytes）;
??????string?out;
??????CHECK（datum.SerializeToString（&out））;
??????txn->Put（caffe::format_int（fileid?*?kCIFARBatchSize?+?itemid?5）?out）;
????}
??}
??txn->Commit（）;
??train_db->Close（）;

??LOG（INFO）?<??scoped_ptr?test_db（db::GetDB（db_type））;
??test_db->Open（output_folder?+?“/cifar10_test_“?+?db_type?db::NEW）;
??txn.reset（test_db->NewTransaction（））;
??//?Open?files
??std::ifstream?data_file（（input_folder?+?“/test_batch.bin“）.c_str（）
??????std::ios::in?|?std::ios::binary）;
??CHECK（data_file）?<??for?（int?itemid?=?0;?itemid?????read_image（&data_file?&label?str_buffer）;
????datum.set_label（label）;
????datum.set_data（str_buffer?kCIFARImageNBytes）;
????string?out;
????CHECK（datum.SerializeToString（&out））;
????txn->Put（caffe::format_int（itemid?5）?out）;
??}
??txn->Commit（）;
??test_db->Close（）;
}

int?main（int?argc?char**?ar

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2016-07-11?05:15??caffe-windows\
?????文件??????101863??2016-07-11?05:15??caffe-windows\.Doxyfile
?????文件????????2518??2016-07-11?05:15??caffe-windows\.gitattributes
?????文件????????1192??2016-07-11?05:15??caffe-windows\.gitignore
?????文件????????1757??2016-07-11?05:15??caffe-windows\.travis.yml
?????文件????????2939??2016-07-11?05:15??caffe-windows\CMakeLists.txt
?????文件????????1917??2016-07-11?05:15??caffe-windows\CONTRIBUTING.md
?????文件?????????620??2016-07-11?05:15??caffe-windows\CONTRIBUTORS.md
?????文件?????????210??2016-07-11?05:15??caffe-windows\INSTALL.md
?????文件????????2095??2016-07-11?05:15??caffe-windows\LICENSE
?????文件???????23516??2016-07-11?05:15??caffe-windows\Makefile
?????文件????????4283??2016-07-11?05:15??caffe-windows\Makefile.config.example
?????文件????????3047??2016-07-11?05:15??caffe-windows\README.md
?????文件?????????725??2016-07-11?05:15??caffe-windows\appveyor.yml
?????文件????????1180??2016-07-11?05:15??caffe-windows\caffe.cloc
?????目錄???????????0??2016-07-11?05:15??caffe-windows\cmake\
?????文件????????4373??2016-07-11?05:15??caffe-windows\cmake\ConfigGen.cmake
?????文件???????11184??2016-07-11?05:15??caffe-windows\cmake\Cuda.cmake
?????文件????????5564??2016-07-11?05:15??caffe-windows\cmake\Dependencies.cmake
?????目錄???????????0??2016-07-11?05:15??caffe-windows\cmake\External\
?????文件????????1939??2016-07-11?05:15??caffe-windows\cmake\External\gflags.cmake
?????文件????????1719??2016-07-11?05:15??caffe-windows\cmake\External\glog.cmake
?????文件????????1764??2016-07-11?05:15??caffe-windows\cmake\Misc.cmake
?????目錄???????????0??2016-07-11?05:15??caffe-windows\cmake\Modules\
?????文件????????1666??2016-07-11?05:15??caffe-windows\cmake\Modules\FindAtlas.cmake
?????文件????????1545??2016-07-11?05:15??caffe-windows\cmake\Modules\FindGFlags.cmake
?????文件????????1451??2016-07-11?05:15??caffe-windows\cmake\Modules\FindGlog.cmake
?????文件????????6723??2016-07-11?05:15??caffe-windows\cmake\Modules\FindLAPACK.cmake
?????文件????????1119??2016-07-11?05:15??caffe-windows\cmake\Modules\FindLMDB.cmake
?????文件????????1728??2016-07-11?05:15??caffe-windows\cmake\Modules\FindLevelDB.cmake
?????文件????????3250??2016-07-11?05:15??caffe-windows\cmake\Modules\FindMKL.cmake
............此處省略716個文件信息