Untitled

 avatar
unknown
plain_text
2 years ago
508 B
23
Indexable
import urllib.request

MODEL_URL = 'http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v2_coco_2018_03_29.tar.gz'
MODEL_NAME = MODEL_URL.split('/')[-1][:-len('.tar.gz')]
LABEL_MAP_URL = 'https://raw.githubusercontent.com/tensorflow/models/master/research/object_detection/data/mscoco_label_map.pbtxt'

urllib.request.urlretrieve(MODEL_URL, MODEL_NAME + '.tar.gz')
!tar -xvf {MODEL_NAME + '.tar.gz'}
!rm {MODEL_NAME + '.tar.gz'}

urllib.request.urlretrieve(LABEL_MAP_URL, 'label_map.pbtxt')
Editor is loading...