Dmitry Kurtaev dkurt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| cmake_minimum_required(VERSION 3.4.3) | |
| project(test_halide CXX) | |
| include_directories( | |
| ~/Halide/build/include | |
| ) | |
| link_directories( | |
| ~/Halide/build/lib | |
| ) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| { | |
| 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
/ text_tf_graph.py
Last active
October 17, 2017 14:27
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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': |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| cv::Mat img = cv::imread("toucan.jpg"); | |
| cv::resize(img, img, cv::Size(224, 224)); | |
| img.convertTo(img, CV_32F); | |
| cv::subtract(img, cv::Scalar(104.0f, 117.0f, 123.0f), img); | |
| cv::divide(img, cv::Scalar(127.5f, 127.5f, 127.5f), img); | |
| std::vector<cv::Mat> channels(3); | |
| cv::split(img, channels); | |
| cv::Mat blob({1, 3, 224, 224}, CV_32F); | |
| for (int i = 0; i < 3; ++i) { | |
| cv::Mat dst(224, 224, CV_32F, blob.ptr<float>(0, i)); |
dkurt
/ ssd_mobilenet_v1_coco_hat.pbtxt
Created
January 24, 2018 10:58
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| node { | |
| name: "image_tensor" | |
| op: "Placeholder" | |
| } | |
| node { | |
| name: "FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/convolution" | |
| op: "Conv2D" | |
| input: "image_tensor" | |
| input: "FeatureExtractor/MobilenetV1/Conv2d_0/weights" | |
| attr { |
dkurt
/ ssd_mobilenet_v1_coco_2017_11_17.pbtxt
Last active
March 14, 2018 07:58
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| node { | |
| name: "image_tensor" | |
| op: "Placeholder" | |
| attr { | |
| key: "dtype" | |
| value { | |
| type: DT_UINT8 | |
| } | |
| } | |
| attr { |
dkurt
/ frozen_inception_resnet_v1.pbtxt
Created
March 30, 2018 08:21
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # http://answers.opencv.org/s/answers/186571 | |
| node { | |
| name: "input_image" | |
| op: "Placeholder" | |
| attr { | |
| key: "dtype" | |
| value { | |
| type: DT_FLOAT | |
| } | |
| } |
dkurt
/ faster_rcnn_resnet50_coco_2018_01_28.pbtxt
Created
April 6, 2018 12:46
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| node { | |
| name: "image_tensor" | |
| op: "Placeholder" | |
| attr { | |
| key: "dtype" | |
| value { | |
| type: DT_UINT8 | |
| } | |
| } | |
| attr { |
dkurt
/ faster_rcnn_resnet101_coco_2018_01_28.pbtxt
Created
April 6, 2018 12:47
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| node { | |
| name: "image_tensor" | |
| op: "Placeholder" | |
| attr { | |
| key: "dtype" | |
| value { | |
| type: DT_UINT8 | |
| } | |
| } | |
| attr { |
dkurt
/ faster_rcnn_inception_v2_coco_2018_01_28.pbtxt
Created
April 6, 2018 12:50
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| node { | |
| name: "image_tensor" | |
| op: "Placeholder" | |
| attr { | |
| key: "dtype" | |
| value { | |
| type: DT_UINT8 | |
| } | |
| } | |
| attr { |
OlderNewer