Giter Club home page Giter Club logo

wsl's Introduction

Weakly Supervised Localization

Directory Structure

Data

Each image location is presumed to be as follows:
wsl_data_dir / data / id.extension
where,

  • data is specified as argument during run time itself - currently supports rsna and chexpert
  • id each unique study or image is accessed by this label - first column of each csv is Id - confirm in wsl_csv_dir
  • extension is also specified as argument during runtime - it is decoupled from id since each extension requires slightly different loading function

CSVs

wsl_csv_dir / data / file_name.csv
Each csv directory 4 main files - info, train, valid, test
info.csv - first column is Id, the rest are ground truths

  • 0/1 for binary classification, simple integers for regression
  • for binary classification if the 0 and 1 are represent different characterstics, name the column 0_1
    e.g. Male_Female where 0 stands for Male and 1 for Female in the column
  • for a > 1 classes, make a single column with a list of ground-truth labels for each Id
  • todo: include support for multi-class regression and multi-class + multi-label ground truths
    train.csv, valid.csv test.csv - single column csv which contains ids to use for current data split
    original.csv - (optional) csv from which info.csv was derived

Model

Please refer to wsl / wsl / train.py for how the models are named

Summary

Used for storing combined results upon testing of all models - ease of comparison for different architectures and variants

What to change?

Go to wsl / locations.py Use user variable to reflect your name
Add to the if-else bock to add the location of storage - maintainer recommends using full paths to avoid confusion
The paths added here can be called all over the repo, do not use actual paths anywhere else
TODO: make setting this a part of docker setup

How to run?

Login to the machine

Occurs on your remote / local machine

Make a folder A where your repository will reside - $ mkdir folder-A
Go to foler-A - $ cd folder-A
Inside folder-A - $ git clone https://github.com/mehak-agr/wsl.git
Make a docker - $ sudo docker run --gpus all --ipc=host -it -v /home/mehak.aggarwal/mnt/:/data --name m_wsl projectmonai/monai:latest

Occurs inside docker m_wsl

Go to folder-A - $ cd /data/2015P002510/...folder-A
Install wsl as a package - $ pip install -e wsl
To train a debug model to ensure things work fine - $ wsl medinet --debug

You can find a comprehensive list of commands and arguments in wsl/main.py
Your trained model will be here - wsl_model_dir / models /

Upcoming updates

  • UNet for segmentation

wsl's People

Contributors

mehak-agr avatar nishanthta avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.