Skip to content

Add new features, PASCAL examples and pretrained models

Compare
Choose a tag to compare
@yoshitomo-matsubara yoshitomo-matsubara released this 17 Dec 04:30
· 650 commits to main since this release
ba665f7

New features

  • Add wrapped resize to enable specifying interpolation for resize (PR #182)
  • Add wrapped random crop resize to enable specifying interpolation for random crop resize (PR #183)
  • Enable to load ckpt containing only specific module and via URL (PR #187)

New examples and trained models

  • Add examples for PASCAL VOC 2012 (PRs #184, #186)
  • Update README (PR #185)
  • Add model weights of DeepLabv3 with ResNet-50/101 fine-tuned on PASCAL VOC 2012 (Segmentation)
mean IoU global pixelwise acc
DeepLabv3 w/ ResNet-50 80.6 95.7
DeepLabv3 w/ ResNet-101 82.4 96.2

Model implementations are available in torchvision. These model weights are originally pretrained on COCO 2017 dataset (available in torchvision) and then fine-tuned on PASCAL VOC 2012 (Segmentation) dataset.

Minor updates

  • Add a version constant (PR #175)
  • Rename and add functions for ResNet-50 and ResNet-101 (PR #176)
  • Add CITATION file (PR #178)
  • Update version (PRs #174, #188)
  • Update README (PRs #179, #180)