Skip to content

This repository is used to provide a code base for PyTorch-based training framework

License

Notifications You must be signed in to change notification settings

baiti01/CodeBase

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

  1. This repository is used to provide a code base for PyTorch-based training framework. One can easily develop their own model on top of it.
  2. Some dirty codes are still there without debugging.
  3. UNet-based denoising/segmentation examples are provided based on two open-access datasets
    1. Denoising: AAPM LowDose CT Challenge dataset
    2. Segmentation: StructSeg segmentation dataset.

step by step:

  1. customize your own dataset within lib/dataset
  2. customize your own specific training workflow within lib/model
  3. customize your own architecture within lib/model/module

About

This repository is used to provide a code base for PyTorch-based training framework

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages