#include "Frontend.h"
#include "Importer.h"
#include <cmdline/View.h>
#include <memory>
#include <fstream>
#include <cassert>
Go to the source code of this file.
◆ make_frontend()
Definition at line 26 of file Entry.cpp.
27{
29
30 auto frontend = std::make_unique<Frontend>();
31
32
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
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 ¶m)
bool from_txt(std::istream &is, ::caffe::NetParameter ¶m)
References from_bin(), and from_txt().