- 博客(5)
- 收藏
- 关注
原创 数据加载data
import numpy as np import torch import torch.utils.data as data import os from PIL import Image import random from torchvision import transforms def is_image_file(filename): # 定义一个判断是否是图片的函数 return any(filename.endswith(extension) for extension in .
2021-07-26 17:03:47 125
原创 train
from __future__ import print_function import argparse import os import random import torch import torch.nn as nn import torch.backends.cudnn as cudnn import torchvision.utils as vutils import time import numpy as np from numpy import * from data_loader.da.
2021-07-26 17:02:18 180
原创 cv 2 tif图片切割 .py
import cv2 as cv #import matplotlib.pyplot as plt import os def crop_image(image_dir, output_path, size,train_tile_padding,useoverlap=True,r=0.5): # image_dir 批量处理图像文件夹 size 裁剪后的尺寸 # 获取图片路径列表 file_path_list = [] for filename in os.listdir(.
2021-07-26 17:01:37 252
原创 bce_predict阈值分割.py
from __future__ import print_function import torch from PIL import Image import os import numpy as np from tqdm import tqdm import torchvision.transforms as transforms from torchvision.transforms import ToPILImage import time import cv2 # from models.Eca_.
2021-07-26 17:00:46 121
原创 pspnet
import torch import torch.nn.functional as F from torch import nn from torchvision import models from torchsummary import summary import torchvision.models as models #from ..utils import initialize_weights #from ..utils.misc import Conv2dDeformable #f...
2021-07-26 16:59:47 174
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人