Chefkrym

Could anyone please post a sample code of using this TTACH for binary segmentation in PyTorch? Can't seem to get it to work. Thank you all :)

Segformer/Transformer Backbone

1

Anyone used a Segformer backbone/any transformer backbone with Segmentation models? Please help.

`import torch import ttach as tta import timm import numpy as np from PIL import Image import matplotlib.pyplot as plt import cv2 model = torch.load('E:/PhD_Projects/egmentation models/new model weights/UNet_mitb2_thresh0.3.pth') tta_model =...