download-and-build-darknet.py

import os
import sys

os.chdir("darknet")

# imports open cv into the repo and allows it to be used
# os.system("import cv2")

#change makefile to have GPU and OPENCV enabled
# os.system("sed -i '' 's/OPENCV=0/OPENCV=1/' Makefile")
# os.system("sed -i 's/GPU=0/GPU=1/' Makefile")
# os.system("sed -i 's/CUDNN=0/CUDNN=1/' Makefile")
# os.system("sed -i 's/CUDNN_HALF=0/CUDNN_HALF=1/' Makefile")

#make darknet (builds darknet so that you can then use the darknet executable file to run or train object detectors)
os.system("make")

# get pretrained weights for darknet for Yolov3
os.system("wget https://pjreddie.com/media/files/yolov3.weights")
# get pretrained weights for darknet for Yolov4
os.system("wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights")
# Get tf-lite weight for yolov3
# os.system("wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov3.weights")
# Get tf-lite weights for yolov4
os.system("wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights")

# TODO: Download the yolo tiny weights for creating a tf-lite model