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