search

dvlab-research / PFENet

PFENet: Prior Guided Feature Enrichment Network for Few-shot Segmentation (TPAMI).
322 stars 53 forks source link

Preprocessing #45

Closed steveazzolin closed 3 years ago

steveazzolin commented 3 years ago

Hi, I'm reading your code and I have a question:

is there a particular reason for which you decided to implement transformations (in transform.py) by hand, instead of relying on the torch torchvision.transforms package? Do you think it can be a source of delays in the code?

Thanks in advance!

tianzhuotao commented 3 years ago

Thanks for being interested in our work.

We mainly followed the implementations of https://github.com/hszhao/semseg which customizes the transformation functions that might be more suitable for semantic segmentation compared to functions available in ``torchvision.transforms'' at that time.