Skip to content

Instantly share code, notes, and snippets.

View AlbertXiebnu's full-sized avatar

xiexiaobo AlbertXiebnu

View GitHub Profile
@AlbertXiebnu
AlbertXiebnu / objectRecog.cc&env=python
Last active January 20, 2021 05:21
tensorflow c++ api to load model and run. from official code base.|-|&tag=gist
#include <fstream>
#include <vector>
#include "tensorflow/cc/ops/const_op.h"
#include "tensorflow/cc/ops/image_ops.h"
#include "tensorflow/cc/ops/standard_ops.h"
#include "tensorflow/core/framework/graph.pb.h"
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/graph/default_device.h"
#include "tensorflow/core/graph/graph_def_builder.h"
## from github discuss
ngram_filters = [3, 4, 5, 6, 7, 8]
conv_filters = []
for n_gram in ngram_filters:
conv_filters.append(Sequential())
conv_filters[-1].add(Convolution2D(nb_feature_maps, 1, n_gram, wv_sz))
conv_filters[-1].add(MaxPooling2D(poolsize=(nb_tokens - n_gram + 1, 1)))
conv_filters[-1].add(Flatten())