search

dddb11 / MVSS-Net

Unofficial implementation of MVSS-Net (ICCV 2021) with Pytorch including training code.
64 stars 10 forks source link

Edge masks #10

Closed IrumaMegumi closed 1 year ago

IrumaMegumi commented 1 year ago

Can you give me a set of sample images of the original image, groundtruth, edge mask, and label? I don't understand how to generate edge mask.

SunnyHaze commented 1 year ago

image

IrumaMegumi commented 1 year ago

That is, a Sobel or Canny at groundtruth?

IrumaMegumi commented 1 year ago

Or give me codes on processing groundtruth to edge mask?

SunnyHaze commented 1 year ago

https://github.com/dddb11/MVSS-Net/blob/15f723da7f24020fe15a8d4a97e0dc300e0b4af7/datasets/dataset.py#L67

IrumaMegumi commented 1 year ago

Thanks.

SunnyHaze commented 1 year ago

Of course, you can use Canny or Sobel to implement a similar function. The width of the edge musk can greatly affect the results.