search

ozantezcan / BSUV-Net-inference

56 stars 13 forks source link

Why not transforming last 2 channel in data_loader.py (line 138 transform(inp, fr[:, :, :1])? #9

Open winwinteam opened 1 year ago

winwinteam commented 1 year ago

Hi @ozantezcan, Merry Christmas! May I know why not transforming last 2 channel in data_loader.py? 

  def __iter__(self):
                for transform in self.transforms_pre:
                    inp, _ = transform(inp, fr[:, :, :1])

                for transform in self.transforms_post:
                    inp, _ = transform(inp, fr[:, :, :1])

https://github.com/ozantezcan/BSUV-Net-inference/blob/master/utils/data_loader.py (line 138)

Thank you!