Skip to content

Releases: yakhyo/crack-segmentation

release v0.0.1

14 Mar 13:15
Compare
Choose a tag to compare

These models trained with 10 epochs. Model weights saved in f16 format (half tensor).

Model names:

  • ce.pt - model trained with cross entropy loss function
  • dice.pt - model trained with dice loss function
  • dicece.pt - model trained with dice cross entropy loss function
  • focal.pt- model trained with focal loss function