invoker-LL / WSI-finetuning

This is the official repository for our CVPR 2023 paper 'Task-Specific Fine-Tuning via Variational Information Bottleneck for Weakly-Supervised Pathology Whole Slide Image Classification'.
78 stars 14 forks source link

Incomplete code of class `Whole_Slide_Bag_FP_SAVE` in `extract_topK_ROIs.py` #3

Closed caiyu6666 closed 1 year ago

caiyu6666 commented 1 year ago

Hello, thank you for your code. However, I met some problem in Stage-2 (wsi-finetuning with topK).

In extract_topK_ROIs.py line 88,

dataset = Whole_Slide_Bag_FP_SAVE(file_path=patch_file_path,wsi=wsi,select_idx=topk_indexes)

However, the class Whole_Slide_Bag_FP_SAVE doesn't have the keyword argument select_idx. Could you please provide the complete code? Thanks a lot!

MrPeterJin commented 1 year ago

I was encountering the same problem. Is there any update?

invoker-LL commented 1 year ago

Sorry for my version updating mistake. Now I have re-committed the correct version of Whole_Slide_Bag_FP_SAVE.