Datasets
Models
Downloads: 1
Diff of
/drunet/predict_segment.py
[000000]
..
[2824d6]
Switch to side-by-side view
--- a +++ b/drunet/predict_segment.py @@ -0,0 +1,8 @@ +import segment + +if __name__ == '__main__': + Seg = segment.Segmentation() + # start predict + input_dir = r'' # Fill in the image path + save_dir = r'' # fill in save path + Seg.predict_and_save(input_dir, save_dir)