layer { name: "data" type: "Data" top: "data" include { phase: TRAIN } transform_param { mean_value: 69 mean_value: 57 mean_value: 59 mean_value: 59 mean_value: 64 mean_value: 46 mean_value: 0 } data_param { source: "/home/leo/database/train/data/" batch_size: 8 backend: LMDB } } layer { name: "data" type: "Data" top: "data" include { phase: TEST } transform_param { mean_value: 69 mean_value: 57 mean_value: 59 mean_value: 59 mean_value: 64 mean_value: 46 mean_value: 0 } data_param { source: "/home/leo/database/test/data/" batch_size: 4 backend: LMDB } } layer { name: "landsat_scenes" type: "Slice" bottom: "data" top: "landsat_scenes" top: "label" slice_param { slice_point: 6 axis: 1 } } layer { name: "Convolution1" type: "Convolution" bottom: "landsat_scenes" top: "Convolution1" convolution_param { num_output: 30 bias_term: false pad: 0 kernel_size: 1 stride: 1 weight_filler { type: "msra" } } } layer { name: "score" type: "Convolution" bottom: "Convolution1" top: "score" convolution_param { num_output: 5 bias_term: true pad: 0 kernel_size: 1 stride: 1 weight_filler { type: "msra" } bias_filler { type: "constant" } } } layer { name: "loss" type: "SoftmaxWithLoss" bottom: "score" bottom: "label" top: "loss" loss_param { ignore_label: 5 weight_by_label_freqs: true class_weighting: 0.064 class_weighting: 0.21 class_weighting: 2.12 class_weighting: 1.95 class_weighting: 1 } } layer { name: "accuracy" type: "Accuracy" bottom: "score" bottom: "label" top: "accuracy" include { phase: TEST } accuracy_param { ignore_label: 5 } } layer { type: 'Python' name: 'evaluation' top: 'IoU' bottom: 'score' bottom: 'label' python_param { module: 'python_confmat' layer: 'PythonConfMat' param_str: '{"test_iter":10000}' } include { phase: TEST } }