Reformat code.
This commit is contained in:
@@ -1,15 +1,13 @@
|
||||
#include <algorithm>
|
||||
#include <cstdio>
|
||||
#include <iostream>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include "Options.hpp"
|
||||
|
||||
using namespace fmri;
|
||||
using namespace std;
|
||||
|
||||
static void show_help(const char* progname, int exitcode)
|
||||
{
|
||||
static void show_help(const char *progname, int exitcode) {
|
||||
cerr << "Usage: " << progname << " -m MODEL -w WEIGHTS INPUTS..." << endl
|
||||
<< endl
|
||||
<< R"END(
|
||||
@@ -24,16 +22,14 @@ Options:
|
||||
exit(exitcode);
|
||||
}
|
||||
|
||||
static void check_file(const char * filename)
|
||||
{
|
||||
static void check_file(const char *filename) {
|
||||
if (access(filename, R_OK) != 0) {
|
||||
perror(filename);
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
Options Options::parse(const int argc, char * const argv[])
|
||||
{
|
||||
Options Options::parse(const int argc, char *const argv[]) {
|
||||
string model;
|
||||
string weights;
|
||||
|
||||
@@ -88,21 +84,17 @@ Options Options::parse(const int argc, char * const argv[])
|
||||
Options::Options(string &&model, string &&weights, vector<string> &&inputs) noexcept:
|
||||
modelPath(move(model)),
|
||||
weightsPath(move(weights)),
|
||||
inputPaths(move(inputs))
|
||||
{
|
||||
inputPaths(move(inputs)) {
|
||||
}
|
||||
|
||||
const string& Options::model() const
|
||||
{
|
||||
const string &Options::model() const {
|
||||
return modelPath;
|
||||
}
|
||||
|
||||
const string& Options::weights() const
|
||||
{
|
||||
const string &Options::weights() const {
|
||||
return weightsPath;
|
||||
}
|
||||
|
||||
const vector<string>& Options::inputs() const
|
||||
{
|
||||
const vector<string> &Options::inputs() const {
|
||||
return inputPaths;
|
||||
}
|
||||
|
||||
@@ -8,13 +8,14 @@ namespace fmri {
|
||||
using std::vector;
|
||||
using std::string;
|
||||
|
||||
class Options
|
||||
{
|
||||
class Options {
|
||||
public:
|
||||
static Options parse(const int argc, char *const argv[]);
|
||||
|
||||
const string &model() const;
|
||||
|
||||
const string &weights() const;
|
||||
|
||||
const vector<string> &inputs() const;
|
||||
|
||||
private:
|
||||
|
||||
@@ -1,5 +1,4 @@
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include "Simulator.hpp"
|
||||
|
||||
using namespace caffe;
|
||||
@@ -7,12 +6,10 @@ using namespace std;
|
||||
using namespace fmri;
|
||||
|
||||
Simulator::Simulator(const string &model_file, const string &weights_file) :
|
||||
net(model_file, TEST)
|
||||
{
|
||||
net(model_file, TEST) {
|
||||
net.CopyTrainedLayersFrom(weights_file);
|
||||
}
|
||||
|
||||
void Simulator::simulate(const string& image_file)
|
||||
{
|
||||
void Simulator::simulate(const string &image_file) {
|
||||
cerr << "This is not implemented yet." << endl;
|
||||
}
|
||||
|
||||
@@ -5,8 +5,7 @@
|
||||
using namespace std;
|
||||
using namespace fmri;
|
||||
|
||||
int main(int argc, char * const argv[])
|
||||
{
|
||||
int main(int argc, char *const argv[]) {
|
||||
::google::InitGoogleLogging(argv[0]);
|
||||
|
||||
Options options = Options::parse(argc, argv);
|
||||
|
||||
Reference in New Issue
Block a user