def get_image_label():
# download label
label_url = 'https://s3.amazonaws.com/onnx-model-zoo/synset.txt'
with open(check_exist_or_download(label_url), 'r') as f:
labels = [l.rstrip() for l in f]
image_url = 'https://s3.amazonaws.com/model-server/inputs/kitten.jpg'
img = Image.open(check_exist_or_download(image_url))
return img, labels
def get_image():
# download image
img1 = Image.open(
check_exist_or_download(
'https://angus-doc.readthedocs.io/en/latest/_images/aurelien.jpg'))
img2 = Image.open(
check_exist_or_download(
'https://angus-doc.readthedocs.io/en/latest/_images/gwenn.jpg'))
return img1, img2
def load_dataset():
train_x_url = 'http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz'
train_y_url = 'http://yann.lecun.com/exdb/mnist/train-labels-idx1-ubyte.gz'
valid_x_url = 'http://yann.lecun.com/exdb/mnist/t10k-images-idx3-ubyte.gz'
valid_y_url = 'http://yann.lecun.com/exdb/mnist/t10k-labels-idx1-ubyte.gz'
train_x = read_image_file(check_exist_or_download(train_x_url)).astype(
np.float32)
train_y = read_label_file(check_exist_or_download(train_y_url)).astype(
np.float32)
valid_x = read_image_file(check_exist_or_download(valid_x_url)).astype(
np.float32)
valid_y = read_label_file(check_exist_or_download(valid_y_url)).astype(
np.float32)
return train_x, train_y, valid_x, valid_y
def load_vocab():
url = 'https://storage.googleapis.com/bert_models/2018_10_18/uncased_L-12_H-768_A-12.zip'
download_dir = '/tmp/'
filename = os.path.join(download_dir, 'uncased_L-12_H-768_A-12', '.',
'vocab.txt')
with zipfile.ZipFile(check_exist_or_download(url), 'r') as z:
z.extractall(path=download_dir)
return filename
def get_image_labe():
labels = [
'neutral', 'happiness', 'surprise', 'sadness', 'anger', 'disgust',
'fear', 'contempt'
]
# download image
image_url = 'https://microsoft.github.io/onnxjs-demo/img/fear.8d1417fa.jpg'
img = Image.open(check_exist_or_download(image_url))
return img, labels
def get_image():
image_url = 'https://raw.githubusercontent.com/simo23/tinyYOLOv2/master/person.jpg'
img = Image.open(check_exist_or_download(image_url))
img = img.resize((416, 416))
return img
def get_image():
# download image
image_url = 'https://s3.amazonaws.com/model-server/inputs/kitten.jpg'
img = Image.open(check_exist_or_download(image_url))
return img
