Skip to content

Latest commit

 

History

History
34 lines (32 loc) · 666 Bytes

File metadata and controls

34 lines (32 loc) · 666 Bytes

transform_pytorch_for_tensorflow

Make Pytorch transforms function working with Tensorflow

List transforms supported

  • Compose
  • ToTensor
  • ToPILImage
  • Normalize
  • Resize
  • Scale
  • CenterCrop
  • Pad
  • Lambda
  • RandomApply
  • RandomChoice
  • RandomOrder
  • RandomCrop
  • RandomHorizontalFlip
  • RandomVerticalFlip
  • RandomResizedCrop
  • RandomSizedCrop
  • FiveCrop
  • TenCrop
  • LinearTransformation
  • ColorJitter
  • RandomRotation
  • RandomAffine
  • Grayscale
  • RandomGrayscale
  • RandomPerspective
  • RandomErasing

Next stage

Convert all method to pure Tensorflow