Giter VIP home page Giter VIP logo

blood-vessel-segmentation-public's Introduction

SenNet + HOA - Hacking the Human Vasculature in 3D solution

https://www.kaggle.com/competitions/blood-vessel-segmentation/overview

PyTorch Lightning Config: Hydra Template

Description

Codebase for SenNet + HOA - Hacking the Human Vasculature in 3D competition (4th Place Solution).

The detailed description is here: https://www.kaggle.com/competitions/blood-vessel-segmentation/discussion/475052

Download data

kaggle competitions download -c blood-vessel-segmentation

Download additional data from the

http://human-organ-atlas.esrf.eu

If you want to skip the initial training without pseudo, you can download the images with pseudo-labels from Kaggle.

kaggle datasets download -d igorkrashenyi/50um-ladaf-2020-31-kidney-pag-0-01-0-02-jp2

Setting up the environment

# clone project
git clone https://github.com/burnmyletters/blood-vessel-segmentation-public
cd blood-vessel-segmentation-public

# [OPTIONAL] create conda environment
conda create -n bvs python=3.10.9
conda activate bvs

# install requirements
pip install -r requirements.txt

How to run

Train model with default configuration

Note: don't forget to modify data_path in bash sripts and in generate_mutliview and generate_mutliview_pseudo files.

# generate multiview
cd scripts
python generate_mutliview.py

# train on base model
sh ./train.sh

# generate pseudos
cd src/utils
python generate_pseudo.py

# train on 3d model with pseudo
sh ./train_pseudo_3d.sh

# train on 2d model with pseudo
sh ./train_pseudo_v2.sh

# Note: if you want to train the best setup which was not submitted to the cometition run 
sh ./train_pseudo.sh

blood-vessel-segmentation-public's People

Contributors

burnmyletters 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.