from experiment import Experiment
from datasets import DroneDeployDataset
from util import limit_memory, enable_dynamic_memory_growth
from model_backends import UnetBaselineModelBackend
enable_dynamic_memory_growth()
# Testing whole pipeline
if __name__ == '__main__':
# Use sample dataset for testing the entire routine
dataset = 'dataset-sample' # 0.5 GB download
dataset = 'dataset-medium' # 9.0 GB download
dataset = DroneDeployDataset(dataset, chip_size=320)\
.download()\
.generate_chips()
model_backend = UnetBaselineModelBackend(320)
experiment = Experiment("test", dataset, model_backend, batch_size=1)
experiment.analyze()
experiment.train(epochs=2)
experiment.save_model()
experiment.generate_inference_test_files()
experiment.score()
experiment.bundle()
print("Experiment finished")
