--- a +++ b/drunet/train_segment.py @@ -0,0 +1,6 @@ +import segment + +if __name__ == '__main__': + Seg = segment.Segmentation() + # start training + Seg.train()