12 lines
268 B
Python
12 lines
268 B
Python
|
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()
|