PaddleX/tutorials/slim/prune/semantic_segmentation/slim_visualize.py

import paddlex as pdx
model = pdx.load_model('output/unet/best_model')
pdx.slim.visualize(model, 'unet.sensi.data', save_dir='./')