os.chdir('picsellia')
if 'api_token' not in os.environ:
raise AuthenticationError("You must set an api_token to run this image")
api_token = os.environ['api_token']
if "host" not in os.environ:
host = "https://app.picsellia.com/sdk/v2/"
else:
host = os.environ["host"]
if "experiment_id" in os.environ:
experiment_id = os.environ['experiment_id']
experiment = Client.Experiment(api_token=api_token, host=host)
exp = experiment.checkout(experiment_id, tree=True, with_file=True)
else:
if "experiment_name" in os.environ and "project_token" in os.environ:
project_token = os.environ['project_token']
experiment_name = os.environ['experiment_name']
experiment = Client.Experiment(api_token=api_token, project_token=project_token, host=host)
exp = experiment.checkout(experiment_name, tree=True, with_file=True)
else:
raise AuthenticationError("You must either set the experiment id or the project token + experiment_name")
experiment.dl_annotations()
experiment.dl_pictures()
experiment.generate_labelmap()
experiment.log('labelmap', experiment.label_map, 'labelmap', replace=True)
import os
from picsellia.client import Client
from picsellia.pxl_exceptions import AuthenticationError
from picsellia_tf2 import pxl_utils
from picsellia_tf2 import pxl_tf
host = 'http://127.0.0.1:8000/sdk/v2/'
api_token = os.environ["api_token"]
experiment_id = os.environ['experiment_id']
project_token = os.environ['project_token']
experiment = Client.Experiment(api_token=api_token,
project_token=project_token,
interactive=False,
host=host)
exp = experiment.checkout(id=experiment_id, tree=True, with_file=True)
experiment.start_logging_chapter('Create records')
experiment.dl_annotations()
experiment.dl_pictures()
experiment.generate_labelmap()
experiment.log('labelmap', experiment.label_map, 'labelmap', replace=True)
experiment.train_test_split()
train_split = {
'x': experiment.categories,
'y': experiment.train_repartition,
'image_list': experiment.train_list_id
}