How to use a pretrained segmentation network as the segmentation network

vicoslab / mixed-segdec-net-comind2021

Official PyTorch implementation for "Mixed supervision for surface-defect detection: from weakly to fully supervised learning"

Other

290 stars 89 forks source link

How to use a pretrained segmentation network as the segmentation network #24

Open Brym-Gyimah opened 2 years ago

Brym-Gyimah commented 2 years ago

Hello, I have been trying to use a pretrained segmentation network like UNET, deeplabv3 or deeplabv3plus inplace of the segmentation network in this paper but cannot seem to go by it. Can I get any help from this community?