ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Entry.cpp File Reference
#include "Frontend.h"
#include "Importer.h"
#include <cmdline/View.h>
#include <memory>
#include <fstream>
#include <cassert>

Go to the source code of this file.

Functions

std::unique_ptr< enco::Frontendmake_frontend (const cmdline::View &cmdline)
 

Function Documentation

◆ make_frontend()

std::unique_ptr< enco::Frontend > make_frontend ( const cmdline::View cmdline)

Definition at line 26 of file Entry.cpp.

27{
28 assert(cmdline.size() == 2);
29
30 auto frontend = std::make_unique<Frontend>();
31
32 // Fill prototxt
33 {
34 std::ifstream ifs{cmdline.at(0)};
35 if (!ifs.is_open())
36 {
37 throw std::runtime_error("Prototxt file open fail");
38 }
39
40 if (!from_txt(ifs, *frontend->prototxt()))
41 {
42 throw std::runtime_error("Filling prototxt fail");
43 }
44 }
45
46 // Fill caffemodel
47 {
48 std::ifstream ifs{cmdline.at(1), std::ios::binary};
49 if (!ifs.is_open())
50 {
51 throw std::runtime_error("Caffemodel file open fail");
52 }
53
54 if (!from_bin(ifs, *frontend->caffemodel()))
55 {
56 throw std::runtime_error("Filling caffemodel fail");
57 }
58 }
59
60 return std::move(frontend);
61}
bool from_bin(std::istream &is, ::caffe::NetParameter &param)
Definition Importer.cpp:35
bool from_txt(std::istream &is, ::caffe::NetParameter &param)
Definition Importer.cpp:23
Definition View.h:23

References from_bin(), and from_txt().