mos2_segmentation

Usage

• Trianing & Test
  python train.py -tr [path to traning data] -ta [path to target data] -e [epoch数] -b [バッチサイズ] -o [path to output] -m [model number] -w [0 or 1] -bi [0 or 1]

• Cross Validation
  python train.py -tr [path to traning data] -ta [path to target data] -e [epoch数] -b [バッチサイズ] -o [path to output] -at [0,1,2,3] -bi [0 or 1]

• model
  0:fcn32s, 1:unet, 2:unet2, 3:pix2pix

Sample

• mos2
  python train.py -e 100 -o ./result/ -ta ./data/label_012/ -g 0 -m 2 -w 0
• graphen
  python train_graphen_gray.py -ta ../data/graphen/label_12/ -tr ../data/graphen/original_halfsize/ -o result/val/ -e 30 -b 1 -l 0.0001 -g 1 -at 1