text_plain_output_1.png
from torchvision import datasets, models, transforms

from torchvision.datasets import ImageFolder

import torch
from torchvision import datasets, models, transforms
import urllib
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
from torch.autograd import Variable
from collections import OrderedDict
from torch.utils.data import Dataset, DataLoader
import cv2
from os import listdir
from torchvision.datasets import ImageFolder
import sys
import time
import torchvision
preprocess = transforms.Compose([transforms.Resize((256, 256)), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
ImageFolder('training', transform=preprocess)
ImageFolder('testing', transform=preprocess)