Lunatic-GPT / BYOL_Med_Pretrain Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

A simple framework to pretrain medical images with BYOL

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.idea		.idea
__pycache__		__pycache__
backbone		backbone
deploy		deploy
.gitattributes		.gitattributes
README.md		README.md
byol_1channel.py		byol_1channel.py
train.py		train.py
utils.py		utils.py

Repository files navigation

BYOL_Med_Pretrain

A simple framework to pretrain medical images with BYOL, inspired by this reop.

difference

medical image support

support loading dcm/mhd/nii directly with SimpleITK
support window apply to images

dual crop

crop1: crop in dataloader to get the same size
crop2: data aug

data aug

medical image friendly data aug (for small ROI)

About

A simple framework to pretrain medical images with BYOL

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%