├── .gitignore ├── LICENSE ├── README.md ├── baselines ├── Attention.py ├── CLAMSB.py ├── MS2MIL.py ├── MaxPooling.py └── TransMIL.py ├── main.py ├── models └── r2gen.py ├── modules ├── att_model.py ├── caption_model.py ├── dataloaders.py ├── datasets.py ├── encoder_decoder.py ├── loss.py ├── metrics.py ├── optimizers.py ├── tokenizers.py ├── trainer.py └── utils.py ├── ocr ├── dataset_csv │ └── splits_0.csv └── pdf2text.py └── pics ├── DPT.png ├── dataset.png └── framework.png /.gitignore: -------------------------------------------------------------------------------- 1 | *pth 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/README.md -------------------------------------------------------------------------------- /baselines/Attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/baselines/Attention.py -------------------------------------------------------------------------------- /baselines/CLAMSB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/baselines/CLAMSB.py -------------------------------------------------------------------------------- /baselines/MS2MIL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/baselines/MS2MIL.py -------------------------------------------------------------------------------- /baselines/MaxPooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/baselines/MaxPooling.py -------------------------------------------------------------------------------- /baselines/TransMIL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/baselines/TransMIL.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/main.py -------------------------------------------------------------------------------- /models/r2gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/models/r2gen.py -------------------------------------------------------------------------------- /modules/att_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/att_model.py -------------------------------------------------------------------------------- /modules/caption_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/caption_model.py -------------------------------------------------------------------------------- /modules/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/dataloaders.py -------------------------------------------------------------------------------- /modules/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/datasets.py -------------------------------------------------------------------------------- /modules/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/encoder_decoder.py -------------------------------------------------------------------------------- /modules/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/loss.py -------------------------------------------------------------------------------- /modules/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/metrics.py -------------------------------------------------------------------------------- /modules/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/optimizers.py -------------------------------------------------------------------------------- /modules/tokenizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/tokenizers.py -------------------------------------------------------------------------------- /modules/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/trainer.py -------------------------------------------------------------------------------- /modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/modules/utils.py -------------------------------------------------------------------------------- /ocr/dataset_csv/splits_0.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/ocr/dataset_csv/splits_0.csv -------------------------------------------------------------------------------- /ocr/pdf2text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/ocr/pdf2text.py -------------------------------------------------------------------------------- /pics/DPT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/pics/DPT.png -------------------------------------------------------------------------------- /pics/dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/pics/dataset.png -------------------------------------------------------------------------------- /pics/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpystan/Wsi-Caption/HEAD/pics/framework.png --------------------------------------------------------------------------------