├── README.md ├── dataset ├── annotations │ ├── 01.xml │ ├── 04.xml │ ├── 05.xml │ ├── 06.xml │ ├── 09.xml │ ├── 11.xml │ ├── 13.xml │ ├── 16.xml │ ├── 17.xml │ ├── 18.xml │ ├── 20.xml │ ├── A01.xml │ ├── A02.xml │ ├── A03.xml │ ├── A04.xml │ ├── A05.xml │ ├── A06.xml │ ├── A07.xml │ ├── A08.xml │ ├── A09.xml │ ├── A10.xml │ ├── test1.xml │ ├── test2.xml │ ├── test3.xml │ ├── test5.xml │ ├── test6.xml │ ├── test7.xml │ └── test8.xml └── data.txt ├── imgs ├── Overview.png ├── annotation.png ├── attention.png └── test.png ├── log └── finetune.log ├── models ├── gcn_lib │ ├── __init__.py │ ├── pos_embed.py │ ├── torch_edge.py │ ├── torch_nn.py │ └── torch_vertex.py ├── graph_vnet.py ├── pos_embed.py └── vision_transformer.py ├── preprocessing ├── filter.py ├── preprocess_finetune.py ├── preprocess_pretrain.py └── preprocess_pretraining.py ├── pretrain ├── dataloader.py ├── imgs │ ├── 1.png │ ├── 2.png │ ├── 3.png │ ├── 4.png │ ├── 5.png │ ├── 6.png │ ├── 7.png │ └── 8.png ├── pretrain.py ├── utils.py ├── vision_transformer.py └── visualize_attention.py ├── requirements.txt ├── test.py ├── train.py ├── utils ├── evaluation.py ├── plot.py ├── pos_embed.py ├── test_loader.py └── train_loader.py └── weight └── checkpoint.txt /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/README.md -------------------------------------------------------------------------------- /dataset/annotations/01.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/01.xml -------------------------------------------------------------------------------- /dataset/annotations/04.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/04.xml -------------------------------------------------------------------------------- /dataset/annotations/05.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/05.xml -------------------------------------------------------------------------------- /dataset/annotations/06.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/06.xml -------------------------------------------------------------------------------- /dataset/annotations/09.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/09.xml -------------------------------------------------------------------------------- /dataset/annotations/11.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/11.xml -------------------------------------------------------------------------------- /dataset/annotations/13.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/13.xml -------------------------------------------------------------------------------- /dataset/annotations/16.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/16.xml -------------------------------------------------------------------------------- /dataset/annotations/17.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/17.xml -------------------------------------------------------------------------------- /dataset/annotations/18.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/18.xml -------------------------------------------------------------------------------- /dataset/annotations/20.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/20.xml -------------------------------------------------------------------------------- /dataset/annotations/A01.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A01.xml -------------------------------------------------------------------------------- /dataset/annotations/A02.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A02.xml -------------------------------------------------------------------------------- /dataset/annotations/A03.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A03.xml -------------------------------------------------------------------------------- /dataset/annotations/A04.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A04.xml -------------------------------------------------------------------------------- /dataset/annotations/A05.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A05.xml -------------------------------------------------------------------------------- /dataset/annotations/A06.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A06.xml -------------------------------------------------------------------------------- /dataset/annotations/A07.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A07.xml -------------------------------------------------------------------------------- /dataset/annotations/A08.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A08.xml -------------------------------------------------------------------------------- /dataset/annotations/A09.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A09.xml -------------------------------------------------------------------------------- /dataset/annotations/A10.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/A10.xml -------------------------------------------------------------------------------- /dataset/annotations/test1.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/test1.xml -------------------------------------------------------------------------------- /dataset/annotations/test2.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/test2.xml -------------------------------------------------------------------------------- /dataset/annotations/test3.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/test3.xml -------------------------------------------------------------------------------- /dataset/annotations/test5.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/test5.xml -------------------------------------------------------------------------------- /dataset/annotations/test6.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/test6.xml -------------------------------------------------------------------------------- /dataset/annotations/test7.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/test7.xml -------------------------------------------------------------------------------- /dataset/annotations/test8.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/dataset/annotations/test8.xml -------------------------------------------------------------------------------- /dataset/data.txt: -------------------------------------------------------------------------------- 1 | BACH dataset: https://zenodo.org/record/3632035 -------------------------------------------------------------------------------- /imgs/Overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/imgs/Overview.png -------------------------------------------------------------------------------- /imgs/annotation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/imgs/annotation.png -------------------------------------------------------------------------------- /imgs/attention.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/imgs/attention.png -------------------------------------------------------------------------------- /imgs/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/imgs/test.png -------------------------------------------------------------------------------- /log/finetune.log: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gcn_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/gcn_lib/__init__.py -------------------------------------------------------------------------------- /models/gcn_lib/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/gcn_lib/pos_embed.py -------------------------------------------------------------------------------- /models/gcn_lib/torch_edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/gcn_lib/torch_edge.py -------------------------------------------------------------------------------- /models/gcn_lib/torch_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/gcn_lib/torch_nn.py -------------------------------------------------------------------------------- /models/gcn_lib/torch_vertex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/gcn_lib/torch_vertex.py -------------------------------------------------------------------------------- /models/graph_vnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/graph_vnet.py -------------------------------------------------------------------------------- /models/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/pos_embed.py -------------------------------------------------------------------------------- /models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/models/vision_transformer.py -------------------------------------------------------------------------------- /preprocessing/filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/preprocessing/filter.py -------------------------------------------------------------------------------- /preprocessing/preprocess_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/preprocessing/preprocess_finetune.py -------------------------------------------------------------------------------- /preprocessing/preprocess_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/preprocessing/preprocess_pretrain.py -------------------------------------------------------------------------------- /preprocessing/preprocess_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/preprocessing/preprocess_pretraining.py -------------------------------------------------------------------------------- /pretrain/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/dataloader.py -------------------------------------------------------------------------------- /pretrain/imgs/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/1.png -------------------------------------------------------------------------------- /pretrain/imgs/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/2.png -------------------------------------------------------------------------------- /pretrain/imgs/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/3.png -------------------------------------------------------------------------------- /pretrain/imgs/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/4.png -------------------------------------------------------------------------------- /pretrain/imgs/5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/5.png -------------------------------------------------------------------------------- /pretrain/imgs/6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/6.png -------------------------------------------------------------------------------- /pretrain/imgs/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/7.png -------------------------------------------------------------------------------- /pretrain/imgs/8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/imgs/8.png -------------------------------------------------------------------------------- /pretrain/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/pretrain.py -------------------------------------------------------------------------------- /pretrain/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/utils.py -------------------------------------------------------------------------------- /pretrain/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/vision_transformer.py -------------------------------------------------------------------------------- /pretrain/visualize_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/pretrain/visualize_attention.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/train.py -------------------------------------------------------------------------------- /utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/utils/evaluation.py -------------------------------------------------------------------------------- /utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/utils/plot.py -------------------------------------------------------------------------------- /utils/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/utils/pos_embed.py -------------------------------------------------------------------------------- /utils/test_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/utils/test_loader.py -------------------------------------------------------------------------------- /utils/train_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lyhkevin/Graph-V-Net/HEAD/utils/train_loader.py -------------------------------------------------------------------------------- /weight/checkpoint.txt: -------------------------------------------------------------------------------- 1 | https://drive.google.com/file/d/1W23nWdyUb-D78yEqWpQwrV-VbW41hDbU/view?usp=share_link --------------------------------------------------------------------------------