| 151 | |
| 152 | template <class Model> |
| 153 | void Segmentor<Model>::LoadWeight(std::string weight_path){ |
| 154 | torch::load(model, weight_path); |
| 155 | model->eval(); |
| 156 | return; |
| 157 | } |
| 158 | |
| 159 | template <class Model> |
| 160 | void Segmentor<Model>::Predict(cv::Mat image, std::string which_class){ |