Prashant Dandriyal pra-dan

💭

Seeking Edge AI opportunities

AI Engineer

pra-dan / tf_to_tflite.sh

Last active April 16, 2020 09:05

	!pip install tensorflow-gpu==1.15.0

	# Convert
	!toco --graph_def_file /content/yolo-v2-tiny-coco.pb \
	--output_file yolo-v2-tiny-coco.tflite \
	--output_format TFLITE \
	--inference_type FLOAT \
	--inference_input_type FLOAT \
	--input_arrays input \
	--output_arrays output

pra-dan / print_first_last_layers_graph.py

Last active April 16, 2020 06:51

	# This script will help print the inputs and output layer names of Tensorflow graph files (.pb)

	import tensorflow as tf
	gf = tf.GraphDef()
	m_file = open('/content/frozen_darknet_yolov3_model.pb','rb')
	gf.ParseFromString(m_file.read())

	with open('somefile.txt', 'a') as the_file:
	for n in gf.node:
	the_file.write(n.name+'\n')

pra-dan / TIDL_frozen_model_to_optimized.sh

Last active April 14, 2020 17:27

Demo code for "Importing TensorFlow Models" section of TIDL setup, mentioned here: "http://software-dl.ti.com/processor-sdk-linux/esd/docs/05_03_00_07/linux/Foundational_Components_TIDL.html#importing-tensorflow-models"

	# Obtain frozen graph of mobilenet v1
	!wget "http://download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224.tgz"
	!tar -xf mobilenet_v1_1.0_224.tgz

	# Get TensorFlow source code repo
	!git clone "https://github.com/tensorflow/tensorflow.git"
	# Optimize graph using TensorFlow tool
	!python "/content/tensorflow/tensorflow/python/tools/optimize_for_inference.py" --input=mobilenet_v1_1.0_224_frozen.pb --output=mobilenet_v1_1.0_224_final.pb --input_names=input --output_names="MobilenetV1/Predictions/Softmax"

pra-dan / test.cpp

Last active March 11, 2020 10:13

	//Tencent is pleased to support the open source community by making FeatherCNN available.

	//Copyright (C) 2018 THL A29 Limited, a Tencent company. All rights reserved.

	//Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
	//in compliance with the License. You may obtain a copy of the License at
	//
	//https://opensource.org/licenses/BSD-3-Clause
	//
	//Unless required by applicable law or agreed to in writing, software distributed