Darknet2TF_using_Darkflow.sh

# Install TensorFlow 1.0
!pip install tensorflow==1.0

# Install dependencies of the DarkFlow
!apt-get update
!pip3 install numpy
!apt-get install python-opencv -y
!pip install cython

# Clean up the directory
%cd /content/
%rm -rf darkflow
# Clone the darkflow repository
!git clone https://github.com/thtrieu/darkflow
# Change into the darkflow dir and install darkflow with pip
%cd darkflow
!python setup.py build_ext --inplace

# Setup directories
%cd /content/darkflow
%rm -rf weights
%mkdir weights
%cd weights
%mkdir built_graph

# Fix the assertation issue by modifying the loader.py.
%cd /content/darkflow
!sed -i "s/self.offset = 16/self.offset = 20/g"  ./darkflow/utils/loader.py

# Get the weigths and .cfg file
!wget -O "cfg/yolo-v2-tiny.cfg" "https://raw.githubusercontent.com/pjreddie/darknet/master/cfg/yolov2-tiny-voc.cfg"
!wget -O "weights/yolo-v2-tiny.weights" "https://pjreddie.com/media/files/yolov2-tiny-voc.weights"

# IMPORTANT: Replace content of lable.txt file, as specified above. ONLY then proceed
print("\nREMOVE ME IF YOU FOLLOWED THE ABOVE STEP :)")

# Convert
!./flow --model cfg/yolo-v2-tiny.cfg --load weights/yolo-v2-tiny.weights --savepb

## You'll find the converted file (.pb) in "content/darkflow/"