Skip to content

Instantly share code, notes, and snippets.

View qinyao-he's full-sized avatar

Qinyao He qinyao-he

  • San Francisco Bay Area
View GitHub Profile
@qinyao-he
qinyao-he / mnist.py
Created August 8, 2018 19:35
tensorflow mnist classification model
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import numpy as np
import tensorflow as tf
tf.logging.set_verbosity(tf.logging.INFO)
@qinyao-he
qinyao-he / tensorrt.py
Last active June 19, 2020 08:28
Optimize frozen tensorflow graph using TensorRT
import os
import tensorflow as tf
import tensorflow.contrib.tensorrt as trt
def get_frozen_graph(graph_file):
"""Read Frozen Graph file from disk."""
with tf.gfile.FastGFile(graph_file, "rb") as f:
graph_def = tf.GraphDef()
@qinyao-he
qinyao-he / mnist_conv_deconv.py
Last active June 21, 2018 19:26
tensorflow mnist conv-deconv
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import numpy as np
import tensorflow as tf
tf.logging.set_verbosity(tf.logging.INFO)