Files downloader for TFOD

download_files.py

# urls
download_urls = {
  'protobuff': "https://github.com/protocolbuffers/protobuf/releases/download/v3.19.4/protoc-3.19.4-win64.zip",
  'tensorflow_models': "https://github.com/tensorflow/models/archive/refs/heads/master.zip",
  'pretrained_model': "http://download.tensorflow.org/models/object_detection/tf2/20200711/ssd_mobilenet_v2_320x320_coco17_tpu-8.tar.gz"
  }
  
 # Download files
 wget.download(download_urls["protobuff"], paths['protoc_path'])
 wget.download(download_urls["tensorflow_models"], paths['api_path'])
 wget.download(download_urls["pretrained_model"], paths['pretrained_model_path'])  
 
 # Extract Files
 !cd {paths['protoc_path']} && tar -xf protoc-3.19.4-win64.zip
 !cd {paths['api_path']} && tar -xf models-master.zip --strip-components 1
 !cd {paths['pretrained_model_path']} && tar -zxvf {pretrained_model + '.tar.gz'}