Installation - Try it
Via git
pip install brevettiaiTry it in Google Colab hosted Jupyter Notebooks
Build your own
from brevettiai.platform import PlatformAPI, PlatformBackend
from telescope import TelescopeSegmentationJob, TelescopeSegmentationSettings
from brevettiai.data.image import CropResizeProcessor
# login with web api
web = PlatformAPI()
# Find dataset to use
datasets = [next(d for d in web.get_dataset() if d.name == "NeurIPS demo vials train")]
# Build settings object
settings = TelescopeSegmentationSettings(
epochs=50,
model="Lightning",
#image=CropResizeProcessor(output_height=224, output_width=224),
classes=['Cap', 'Thread', 'Neck', 'Container'],
enable_augmentation=False,
)
# Create model on platform
model = web.create_model("NeurIPS segmentation demo", datasets=datasets, settings=settings)
print(f"{web.host}/models/{model.id}")
# Obtain the job object from the web api (provision your machine)
job = web.initialize_training(model, job_type=TelescopeSegmentationJob)
# Start running the job
job.start()Last updated
Was this helpful?