Created
May 1, 2020 15:26
-
-
Save pra-dan/c98c5eeb7482a4bedca3a4f29bb5123e to your computer and use it in GitHub Desktop.
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
# 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