Speed up and using mask image - Githubissues

IyatomiLab / LeafGAN

Other

71 stars 23 forks source link

Speed up and using mask image #5

Closed iron316 closed 3 years ago

iron316 commented 3 years ago

Changes

I add new option that load from mask image (i.e. IMAGE_DIR/trainA_mask )
I add new script that create mask using LFLSeg module

What is advantage?

Learning Leaf GAN is about 3 times faster
We can leaf masking from other model (i.e. DPT)

iron316 commented 3 years ago

I want to review for last check.