fix the problem for --class_balancing

GeorgeSeif / Semantic-Segmentation-Suite

Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!

2.51k stars 880 forks source link

fix the problem for --class_balancing #70

Closed arasharchor closed 6 years ago

arasharchor commented 6 years ago

fix the problem for --class_balancing added tf.reducesum for weighted loss of multiple class segmentation task. added weighted loss for binary segmentation task. added colored package. replaced "checkpoints" with "checkpoints"+args.dataset