Log In Sign Up
Models:
David Feaster
DavidFeaster/
DR-UNet
0
Downloads: 1
[2824d6]: / drunet / predict_segment.py

Download this file

9 lines (7 with data), 236 Bytes 

1
2
3
4
5
6
7
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)