Skip to content

Instantly share code, notes, and snippets.

View dkurt's full-sized avatar

Dmitry Kurtaev dkurt

View GitHub Profile
import tensorflow as tf
# Read the graph.
with tf.gfile.FastGFile('opt_graph.pb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
# Remove Const nodes.
for i in reversed(range(len(graph_def.node))):
if graph_def.node[i].op == 'Const':
@dkurt
dkurt / sample.cpp
Last active August 8, 2017 11:20
sample
{
Halide::Var x("x"), xo("xo"), xi("xi");
Halide::Func f("f");
f(x) = x + 777;
f.bound(x, 0, 16);
f.split(x, xo, xi, 8).gpu_threads(xi).gpu_blocks(xo);
f.output_buffer().dim(0).set_bounds(0, 16);
Halide::Target target = Halide::get_host_target();
@dkurt
dkurt / CMakeLists.txt
Created July 4, 2017 11:08
CMakeLists.txt
cmake_minimum_required(VERSION 3.4.3)
project(test_halide CXX)
include_directories(
~/Halide/build/include
)
link_directories(
~/Halide/build/lib
)
cd ~/opencv/build
cmake -DCMAKE_BUILD_TYPE=RELEASE \
-DWITH_OPENCL=NO \
-DWITH_CUDA=NO \
-DBUILD_TESTS=OFF \
-DBUILD_PERF_TESTS=OFF \
-DBUILD_opencv_aruco=OFF \
-DBUILD_opencv_bgsegm=OFF \
-DBUILD_opencv_bioinspired=OFF \