PSPNET-cudnn5
PSPNET-cudnn5 copied to clipboard
Training script for training PSPNet
Yeah its great to see that PSPNET has updated from cuda7.5 to cuda-8.0 which is very handy to developers. Can you kindly provide scripts on how to train PSPNet model? Kindly do the needful
Grad to see the work really helps you:) PSPNET is based on caffe configure so you can use it just like caffe. Maybe you can read some tutorials about how to use caffe. However, PSPNET provides extra layers which caffe doesn't have:
- densecrf_layer
- domain_transform_forward_only_layer
- domain_transform_layer
- image_seg_data_layer
- interp_layer
- mat_read_layer
- mat_write_layer
- seg_accuracy_layer
- spatial_product_layer
- unique_label_layer
The usage is shown in the PSPNET paper. Or, another way, you can run the evaluation code provided by the author in Matlab which can help you to understand how to use these API quickly.