Skip to content

aenglebert/ortho_vlp_eval

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

67 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Evaluation scripts for "Self-supervised vision-langage alignment of deep learning representations for bone X-rays analysis"

Installation

Install requirements with pip install -r requirements.txt

Configuration

This repository uses hydra configuration files in the "config" folder. You may need to adapt the location of your datasets in the respective configuration files in the config/dataset/ folder.

It uses the following datasets:

  • MURA
  • FracAtlas
  • OAI

You can find pretrained models from the paper here

Classification

To run the classification script, do as follow:

python classification.py dataset=dataset eval_type=eval_type vision_model=vision_model

while replacing "dataset", "eval_type" and "vision_model" by the desired configuration file name for each.

Regression

To run the regression script, do as follow:

python regression.py dataset=dataset eval_type=eval_type vision_model=vision_model

while replacing "dataset", "eval_type" and "vision_model" by the desired configuration file name for each.

nb: The development of the detection script was abandoned, but we've kept it available in case anyone is interested.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages