93fde518ce45a1373c4b635c15e25312b2cc3a0f,examples/semantic_segmentation/fcn/train_fcn.py,,main,#Any#Any#Any#Any#Any#Any#,17

Before Change

def main(gpu=-1, batch_size=1, iterations=100000,
         lr=1e-10, out="result", resume=""):
    // prepare datasets
    wrappers = [lambda d: SubtractWrapper(d),
                lambda d: PadWrapper(
                    d, max_size=(512, 512), preprocess_idx=[0, 1],
                    bg_values={0: 0, 1: -1})]
    train_data = VOCSemanticSegmentationDataset(mode="train")
    test_data = VOCSemanticSegmentationDataset(mode="val")
    for wrapper in wrappers:
        train_data = wrapper(train_data)
        test_data = wrapper(test_data)

    // set up FCN32s
    n_class = 21
    model = FCN32s(n_class=n_class)
    if gpu != -1:
        model.to_gpu(gpu)

After Change

    train_data = VOCSemanticSegmentationDataset(mode="train")
    test_data = VOCSemanticSegmentationDataset(mode="val")
    extend(train_data, transform)
    extend(test_data, transform)

    // set up FCN32s
    n_class = 21

Instances