Skip to content

Instantly share code, notes, and snippets.

@pra-dan
Created May 1, 2020 15:26
Show Gist options
  • Save pra-dan/c98c5eeb7482a4bedca3a4f29bb5123e to your computer and use it in GitHub Desktop.
Save pra-dan/c98c5eeb7482a4bedca3a4f29bb5123e to your computer and use it in GitHub Desktop.
# Install TensorFlow
!pip install tensorflow==1.11.0
# Get repository
!git clone https://github.com/mystic123/tensorflow-yolo-v3.git
%cd tensorflow-yolo-v3
!git checkout ed60b90
# Get coco.names
!wget https://raw.githubusercontent.com/pjreddie/darknet/master/data/coco.names
# Get v3 weights
!wget "https://pjreddie.com/media/files/yolov3.weights"
# Get v3-tiny weights
!wget "https://pjreddie.com/media/files/yolov3-tiny.weights"
# Convert v3
!python3 convert_weights_pb.py --class_names coco.names --data_format NHWC --weights_file yolov3.weights
# Note: The file saves the output frozen graph with the same name : "frozen_darknet_yolov3_model.pb."
# So, make sure to save/rename the result before running the below command
# Convert v3-tiny
!python3 convert_weights_pb.py --class_names coco.names --data_format NHWC --weights_file yolov3-tiny.weights --tiny
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment