mit-han-lab / gan-compression

[CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs
Other
1.1k stars 148 forks source link

Instance Segmentation using pix2pix #74

Closed seekingdeep closed 3 years ago

seekingdeep commented 3 years ago

@lmxyy @junyanz Hi there,

i am trying to use pix2pix to train an instance segmentation model, My input images that are used in training are 256*256 crops of my original 1024*1024 images.

The issue is that:

Possible solution: # 44 is to store the image tensor on the disk and then modify the dataloader to restore the tensors from the disk.

i think using the HD original images 1024*1024 in training should get better results, Waiting for your reply

iteration 185000, Prediction: predition

Groundtruth: 10013

lmxyy commented 3 years ago

I think you could try the original resolution. Storing the image tensor on the disk and then modify the data loader to restore the tensors from the disk is a good solution when running the get_real_stat.py is a good solution and deserves a trial.