unet3 图片预测 动态分辨率

unet3本来不支持动态分辨率，本代码做了修改，可以支持

目录

图片预测代码：

pytorch实现网络代码，不支持447*448这样的分辨率

unet3 修改，支持动态分辨率

---

图片预测代码：

import os
import time

import cv2
from skimage import io, transform
import torch
import torchvision
from torch.autograd import Variable
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import Dataset, DataLoader
from torchvision import transforms  # , utils
# import torch.optim as optim

import numpy as np
from PIL import Image
import glob


from model import U2NETP  # small version u2net 4.7 MB

# normalize the predicted SOD probability map
from model.bisnet import BiSeNetV2
from model.bisnet_s4 import BiSeNetV2_s4
from model.u2net_new import U2NETP