Some questions about dataloader - Githubissues

kaixin96 / PANet

Code for our ICCV 2019 paper PANet: Few-Shot Image Semantic Segmentation with Prototype Alignment

323 stars 64 forks source link

Some questions about dataloader #5

Closed lailvlong closed 4 years ago

lailvlong commented 4 years ago

https://github.com/kaixin96/PANet/blob/c248b2542e2ae1906d11c85b493bc9000773d5fd/dataloaders/common.py#L155 https://github.com/kaixin96/PANet/blob/c248b2542e2ae1906d11c85b493bc9000773d5fd/dataloaders/common.py#L156 https://github.com/kaixin96/PANet/blob/c248b2542e2ae1906d11c85b493bc9000773d5fd/dataloaders/common.py#L157 https://github.com/kaixin96/PANet/blob/c248b2542e2ae1906d11c85b493bc9000773d5fd/dataloaders/common.py#L158 https://github.com/kaixin96/PANet/blob/c248b2542e2ae1906d11c85b493bc9000773d5fd/dataloaders/common.py#L159 In this sample operation, the 'k' is set to be 'n_elements'. I think the 'k' here should be the number of class, but 'n_elements' is the number of images per class. Is it wrong?