Code for SERGCN

Here we present the codes of SERGCN. The paper can be found here.

The codes and the parameters can be downloaded from Google Drive

python version is 3.6.9

torch version is 1.4.0

To train the model on PISC_COARSE dataset, run run_pisc_coarse.sh

To train the model on PISC_FINE dataset, run run_pisc_fine.sh

To train the model on PIPA_COARSE dataset, run run_pipa_coarse.sh

To train the model on PIPA_FINE dataset, run run_pipa_fine.sh

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
.idea		.idea
Save_Model		Save_Model
relation_split		relation_split
utils		utils
.DS_Store		.DS_Store
.gitignore		.gitignore
8168paper.pdf		8168paper.pdf
AverageMeter.py		AverageMeter.py
README.md		README.md
RGCN.py		RGCN.py
SERGCN.py		SERGCN.py
SRDataset.py		SRDataset.py
log_pipa_coarse_9950.txt		log_pipa_coarse_9950.txt
log_pipa_fine_8093.txt		log_pipa_fine_8093.txt
log_pisc_coarse_4813.txt		log_pisc_coarse_4813.txt
log_pisc_fine_7985.txt		log_pisc_fine_7985.txt
main.py		main.py
main_ssl.py		main_ssl.py
resnet_roi.py		resnet_roi.py
run_pipa_coarse.sh		run_pipa_coarse.sh
run_pipa_fine.sh		run_pipa_fine.sh
run_pisc_coarse.sh		run_pisc_coarse.sh
run_pisc_fine.sh		run_pisc_fine.sh

Provide feedback