diff options
Diffstat (limited to 'gantools/ganbreeder.py')
| -rw-r--r-- | gantools/ganbreeder.py | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/gantools/ganbreeder.py b/gantools/ganbreeder.py index b55bc41..a2546a2 100644 --- a/gantools/ganbreeder.py +++ b/gantools/ganbreeder.py @@ -1,10 +1,11 @@ # client functions for interacting with the ganbreeder api import requests import json +import numpy as np def login(username, password): def get_sid(): - url = 'https://ganbreeder.app/login' + url = 'https://artbreeder.com/login' r = requests.get(url) r.raise_for_status() for c in r.cookies: @@ -13,7 +14,7 @@ def login(username, password): return c.value def login_auth(sid, username, password): - url = 'https://ganbreeder.app/login' + url = 'https://artbreeder.com/login' headers = { 'Content-Type': 'application/json', } @@ -34,15 +35,26 @@ def login(username, password): login_auth(sid, username, password) return sid +def parse_info_dict(info): + keyframe = dict() + keyframe['truncation'] = np.float(info['truncation']) + keyframe['latent'] = np.asarray(info['latent']) + classes = info['classes'] + keyframe['label'] = np.zeros(1000)# length of label ("classes") vector: 1000 + for c in info['classes']: + # artbreeder class entries look like [index, value] where index < 1000 + keyframe['label'][c[0]] = c[1] + return keyframe + def get_info(sid, key): if sid == '': raise Exception('Cannot get info; session ID not defined. Be sure to login() first.') cookies = { 'connect.sid': sid } - r = requests.get('http://ganbreeder.app/info?k='+str(key), cookies=cookies) + r = requests.get('http://artbreeder.com/info?k='+str(key), cookies=cookies) r.raise_for_status() - return(r.json()) + return parse_info_dict(r.json()) def get_info_batch(username, password, keys): l = list() |