glog和gflags两个模块的使用实例
2018-01-13 18:28
387 查看
在Caffe中convert_imageset.cpp中有这两个模块的使用,我只是做了一点注释。我觉得学习别人的代码,是一个很好的方法。
算了废话不多说了,直接上代码吧!
算了废话不多说了,直接上代码吧!
// This program converts a set of images to a lmdb/leveldb by storing them // as Datum proto buffers. // Usage: // convert_imageset [FLAGS] ROOTFOLDER/ LISTFILE DB_NAME // // where ROOTFOLDER is the root folder that holds all the images, and LISTFILE // should be a list of files as well as their labels, in the format as // subfolder1/file1.JPEG 7 // .... #include <algorithm> #include <fstream> // NOLINT(readability/streams) #include <string> #include <utility> #include <vector> #include "boost/scoped_ptr.hpp" #include "gflags/gflags.h" #include "glog/logging.h"//增加glog的头文件 #include "caffe/proto/caffe.pb.h" #include "caffe/util/db.hpp" #include "caffe/util/format.hpp" #include "caffe/util/io.hpp" #include "caffe/util/rng.hpp" using namespace caffe; // NOLINT(build/namespaces) using std::pair; using boost::scoped_ptr; -------------------------------------------------------------------------------- 命令行解析参数的定义在主函数的外面,类似全局变量,使用时“FLAGS_参数名” DEFINE_bool(gray, false, "When this option is on, treat images as grayscale ones"); DEFINE_bool(shuffle, false, "Randomly shuffle the order of images and their labels"); DEFINE_string(backend, "lmdb", "The backend {lmdb, leveldb} for storing the result"); DEFINE_int32(resize_width, 0, "Width images are resized to"); DEFINE_int32(resize_height, 0, "Height images are resized to"); DEFINE_bool(check_size, false, "When this option is on, check that all the datum have the same size"); DEFINE_bool(encoded, false, "When this option is on, the encoded image will be save in datum"); DEFINE_string(encode_type, "", "Optional: What type should we encode the image as ('png','jpg',...)."); --------------------------------------------------------------------------- int main(int argc, char** argv) { #ifdef USE_OPENCV --------------------------------------------------------------------------- ::google::InitGoogleLogging(argv[0]);//打印日志初始化 // Print output to stderr (while still logging) FLAGS_alsologtostderr = 1;//也向标准输入设备输出信息,即写入文件,也在显示器上打印 #ifndef GFLAGS_GFLAGS_H_ namespace gflags = google; #endif //下面设置帮助信息 gflags::SetUsageMessage("Convert a set of images to the leveldb/lmdb\n" "format used as input for Caffe.\n" "Usage:\n" " convert_imageset [FLAGS] ROOTFOLDER/ LISTFILE DB_NAME\n" "The ImageNet dataset for the training demo is at\n" " http://www.image-net.org/download-images\n"); gflags::ParseCommandLineFlags(&argc, &argv, true); //解析命令行参数,参数true时,解析完参数后argc=1,argv只保留了argv[0] //参数flase时,解析完参数后argc个数不变,argv顺序会变化 if (argc < 4) {//如果参数个数小于指定值,显示帮助信息 gflags::ShowUsageWithFlagsRestrict(argv[0], "tools/convert_imageset"); return 1; } //参数的使用 const bool is_color = !FLAGS_gray; const bool check_size = FLAGS_check_size; const bool encoded = FLAGS_encoded; const string encode_type = FLAGS_encode_type; -------------------------------------------------------------------------------- std::ifstream infile(argv[2]); std::vector<std::pair<std::string, int> > lines; std::string line; size_t pos; int label; while (std::getline(infile, line)) { pos = line.find_last_of(' '); label = atoi(line.substr(pos + 1).c_str()); lines.push_back(std::make_pair(line.substr(0, pos), label)); } if (FLAGS_shuffle) { // randomly shuffle data LOG(INFO) << "Shuffling data"; shuffle(lines.begin(), lines.end()); } LOG(INFO) << "A total of " << lines.size() << " images."; if (encode_type.size() && !encoded) LOG(INFO) << "encode_type specified, assuming encoded=true."; int resize_height = std::max<int>(0, FLAGS_resize_height); int resize_width = std::max<int>(0, FLAGS_resize_width); // Create new DB scoped_ptr<db::DB> db(db::GetDB(FLAGS_backend)); db->Open(argv[3], db::NEW); scoped_ptr<db::Transaction> txn(db->NewTransaction()); // Storing to db std::string root_folder(argv[1]); Datum datum; int count = 0; int data_size = 0; bool data_size_initialized = false; for (int line_id = 0; line_id < lines.size(); ++line_id) { bool status; std::string enc = encode_type; if (encoded && !enc.size()) { // Guess the encoding type from the file name string fn = lines[line_id].first; size_t p = fn.rfind('.'); if ( p == fn.npos ) LOG(WARNING) << "Failed to guess the encoding of '" << fn << "'"; enc = fn.substr(p); std::transform(enc.begin(), enc.end(), enc.begin(), ::tolower); } status = ReadImageToDatum(root_folder + lines[line_id].first, lines[line_id].second, resize_height, resize_width, is_color, enc, &datum); if (status == false) continue; if (check_size) { if (!data_size_initialized) { data_size = datum.channels() * datum.height() * datum.width(); data_size_initialized = true; } else { const std::string& data = datum.data(); CHECK_EQ(data.size(), data_size) << "Incorrect data field size " << data.size(); } } // sequential string key_str = caffe::format_int(line_id, 8) + "_" + lines[line_id].first; // Put in db string out; CHECK(datum.SerializeToString(&out)); txn->Put(key_str, out); if (++count % 1000 == 0) { // Commit db txn->Commit(); txn.reset(db->NewTransaction()); LOG(INFO) << "Processed " << count << " files."; } } // write the last batch if (count % 1000 != 0) { txn->Commit(); LOG(INFO) << "Processed " << count << " files."; } #else LOG(FATAL) << "This tool requires OpenCV; compile with USE_OPENCV."; #endif // USE_OPENCV return 0; }
相关文章推荐
- 一个使用泛型堆栈模块创建的两个容纳不同类型数据的实例
- Python使用email模块对邮件进行编码和解码的实例教程
- Python使用difflib对比两个文件操作实例
- 一步一步教你使用AgileEAS.NET基础类库进行应用开发-WinForm应用篇-实例一个模块(商品字典)
- 使用Python的MD5模块比较两个文件是否一致
- 使用触发器即时同步两个表的实例
- Python的加密模块md5、sha、crypt使用实例
- nodejs命令行参数处理模块commander使用实例
- Python使用difflib对比两个文件操作实例
- python中的hashlib和base64加密模块使用实例
- python中使用smtplib和email模块发送邮件实例
- 使用Python的urllib和urllib2模块制作爬虫的实例教程
- Python使用urllib2模块抓取HTML页面资源的实例分享
- 在python中的socket模块使用代理实例
- nodejs命令行参数处理模块commander使用实例
- python使用marshal模块序列化实例
- Python使用pygame模块编写俄罗斯方块游戏的代码实例
- matlab/simulink中自定义c-s函数作为simulink模块使用实例
- Python随机生成数模块random使用实例
- Python multiprocessing模块中的Pipe管道使用实例