search

dvlab-research / PFENet

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

The problem about the generation method of dataset #40

Open ElingBetty opened 3 years ago

ElingBetty commented 3 years ago

As seen at the line 60th in dataset.py, did you only select a query image when the number of the object pixels in this image is larger than 2 32 32 ?

ElingBetty commented 3 years ago

I have run your code without the content about removing the line 60th in dataset.py, and I found there is a considerable drop of the performance.

tianzhuotao commented 3 years ago

We have commented in our code that we follow OSLSM (the first paper in few-shot segmentation) to implement the 60th line, and both CANet and PGNet adopt this strategy to screen out the small objects.

Of course, you can make a comparison with other methods by adding this condition following the aforementioned papers.

Also, this line mainly controls the mask sizes of the support samples, and it only slightly affects the performance. You can find more details in this issue: https://github.com/Jia-Research-Lab/PFENet/issues/11.

Thank you.