Commit 1c907195 authored by E144069X's avatar E144069X

Removed useless import

parent 52eeb80f
......@@ -9,7 +9,6 @@ import torchvision
from PIL import Image
import pims
import time
import args
......@@ -22,8 +21,6 @@ logging.getLogger('libav').setLevel(logging.ERROR)
import utils
import formatData
import albumentations
from albumentations import Compose
import digitExtractor
import cv2
......@@ -113,8 +110,6 @@ class SeqTrDataset(torch.utils.data.Dataset):
self.augmentData = augmentData
if augmentData:
raise ValueError("Data augmentation with albumentation has been removed.")
elif gridShuffle:
self.transf = albumentations.RandomGridShuffle(grid=(gridShuffleSize, gridShuffleSize), p=1.0)
else:
self.transf = None
......@@ -246,10 +241,7 @@ class TestLoader():
else:
self.reSizeTorchFunc = None
if gridShuffle:
self.transf = albumentations.RandomGridShuffle(grid=(gridShuffleSize, gridShuffleSize), p=1.0)
else:
self.transf = None
self.transf = None
self.preproc = PreProcess(self.maskTimeOnImage,self.mask,self.origImgSize,self.resizeImage,self.imgSize,\
augmentData=False,gridShuffle=gridShuffle,transfFunc=self.transf,dataset=self.dataset)
......
import pims
import numpy as np
import xml.etree.ElementTree as ET
import os
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment