PSPNET-cudnn5 icon indicating copy to clipboard operation
PSPNET-cudnn5 copied to clipboard

Training script for training PSPNet

Open chowkamlee81 opened this issue 6 years ago • 1 comments

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

chowkamlee81 avatar Nov 09 '17 09:11 chowkamlee81

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.

BassyKuo avatar Nov 17 '17 04:11 BassyKuo