Method SegDataset

lesson6-Segmentation/SegDataset.cpp:93–111

}

SegDataset::SegDataset(int _resize_width, int _resize_height, std::vector<std::string> _list_images,
                           std::vector<std::string> _list_labels, std::vector<std::string> _name_list)
{
    name_list = _name_list;
    resize_width = _resize_width;
    resize_height = _resize_height;
    list_images = _list_images;
    list_labels = _list_labels;
    for(int i=0; i<_name_list.size(); i++){
        name2index.insert(std::pair<std::string, int>(_name_list[i], i));
    }
    std::vector<cv::Scalar> color_list = get_color_list();
    if(_name_list.size()>color_list.size()){
        throw "Num of classes exceeds defined color list, please add color to color list in SegDataset.cpp";
    }
    for(int i = 0; i<_name_list.size(); i++){
        name2color.insert(std::pair<std::string, cv::Scalar>(_name_list[i],color_list[i]));
    }
}

torch::data::Example<> SegDataset::get(size_t index) {
    std::string image_path = list_images.at(index);