from uav_seg import UAVSegTrainer

if __name__ == '__main__':
    trainer = UAVSegTrainer(
        data_dir='data',
        model_type='deeplabv3plus',
        num_classes=7,
        batch_size=8,
        epochs=1,
        learning_rate=0.001
    )
    trainer.train()