Skip to content

Awenbocc/CPCR

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CPCR & CPRD

Code Download

git clone https://github.com/Awenbocc/CPCR.git

Shared Dataset For both Methods

(1) The Pre-training dataset can be downloaded from here

(2) Unzip it to datasets package

CPCR

Pre-training Step: CP

python pre-train/main.py \
  -a resnet50 \
  --lr 0.015 \
  --mlp --moco-t 0.2 --moco-k 4096 --aug-plus --cos\
  --batch-size 128 \
  --dist-url 'tcp://localhost:10001' --multiprocessing-distributed --world-size 1 --rank 0 \
  /dataset/crd

Fine-tuning Step: CR

2023/03/01: code checking, coming soon...

For VQA-RAD:

bash med-vqa/vqa_rad.sh

For Slake:

bash med-vqa/slake.sh

CPRD (Contrastive Pre-training and Representation Distillation for Medical Visual Question Answering Based on Radiology Images)

Pre-training Step: Teacher Pre-training and Student Representation Distillation

Fine-tuning Step:

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published