search

dddb11 / MVSS-Net

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

Add Transforms that apply simultaneously to img, mask and edges #5

Closed kostino closed 1 year ago

kostino commented 1 year ago

This solves the problem of having to generate edges online when applying naive manipulations, as discussed with @SunnyHaze so that we can have the rest of the edge maps made when creating the dataset to save execution time

SunnyHaze commented 1 year ago

Thank for your implementation friend! Recently I am a little bit busy, and if I have spare time, I will split the edge_generator CLASS into "off-line" version (My implementation which could faster on GPU when have large number of images to transform before trainning) and "online version" that come from your implementation (Canny implementation which faster with augmentations of RandomInpainting & RandomCopyMove on CPU). Thank for your PR again!

SunnyHaze commented 1 year ago

I will also re-manage this repo with @dddb11 and finish a README that could make this reproduction contribute to the research sociaty!