Skip to content

More and better Transforms

Latest
Compare
Choose a tag to compare
@MalteEbner MalteEbner released this 29 Nov 08:08
· 7 commits to master since this release
87dde05

Changes

New transforms

Switch to version-independent torchvision transforms.

  • If torchvision transforms v2 are available, they are used. Otherwise torchvision transforms v1 are used. For details see this comment.
  • Add Transform for DetCon + MultiViewTransformV2 for torchvision.transforms.v2 (#1737)

Typing, naming & docstring improvements

Other

  • Fix DINOProjectionHead BatchNorm Handling (#1729)
  • Add masked average pooling for pooling with segmentation masks (DetCon)(#1739)

Many thanks to all of our contributors!

Models