Python读取tif格式文件需要安装libtiff
,此外需要安装 inferno
本文适用于读取三维tif。
from torch.utils.data import DataLoader, Dataset
from inferno.io.transform.base import Transform, Compose
from inferno.io.transform.generic import Normalize, AsTorchBatch
from inferno.io.transform.image import RandomCrop, RandomRotate, RandomFlip
from libtiff import TIFF
import os
import torch
import numpy as np
定义一个MyDataSet
class MyDataSet(Dataset):
def __init__(self, pathLst, transform): # Parameters and their form vary according to program needs
dataPath, labelPath = pathLst
self.tifStreamData, self.tifStreamLabel = [], []
dataFiles,