Skip to content

lihan0705/deeplearning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TODO plan for Unet

  1. config # not a good way to do this

  2. unet from scratch # unet

  3. dataloader # add basic dataset class for segmentations

  4. loss

  5. optimizers

  6. training # 1. train step 2. loader 3. save checkpoint 4. save train and val loss (todo : log)

  7. ViT from scratch

  8. ViT from scratch # qvk not clear

  9. add cityscapes dataset for segmentation, can run but net output and target shape confused

  10. loss function pytorch bug # https://clay-atlas.com/blog/2020/05/16/pytorch-cn-error-runtimeerror-ambiguous/

  11. running successfully!! # total label 35, solve CUDA ERROR: device-side assert triggered at problem, reason:number of classes and network out of classes is not identical. loss

  12. loss is decreasing but very slowly, accurary is not right!,more evaluation.

  13. 200 epoch without too much data arguementation loss

  14. add inference

  15. inference structure and training structure and trace

About

segmentation unet

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published