Skip to content

adheep04/transformer-voc-seg

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

an image segmentation model designed by me using the swin-v2-t transformer classifier as a backbone, a feature pyramid network, lion optimizer, and deep supervised training (using auxiliary layers for loss)

current

tried using boundary refinement module (sobel kernel) to improve IoU score (0.66 on pascal voc 2012) but i think it may have just added noise:

with boundary-module (purple), without (blue)

alt text

older ver

training for swin-based fpn segmentation model: experienced irrecoverable loss at 2k steps. Pixel accuracy: 0.75, miou: 0.25

alt text

alt text

About

custom transformer-based image segmentation model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published