#!/usr/bin/env python3#-*- coding: utf-8 -*-
"""Created on Fri Mar 30 15:46:09 2018
@author: lps"""
importtorchimporttorch.nn as nnimporttorch.nn.functional as Ffrom torch.autograd importVariableimporttorch.optim as optimimporttorchvision.models as modelsimporttorchvisionfrom torch.utils.data importDataset, DataLoaderfrom torchvision importtransforms, utilsimportmatplotlib.pyplot as pltfrom PIL importImage#import pandas as pd
importnumpy as npimportosimportcopy, time
file_path= '/home/lps/yanzm'BATCH_SIZE= 16EPOCH= 10
#Load data
classdataset(Dataset):def __init__(self, root_dir, label_file, transform=None):
self.root_dir=root_dir
self.label=np.loadtxt(label_file)
self.transform=transformdef __getitem__(self, idx):
img_name= os.path.join(self.root_dir,'%.4d.jpg'%idx)
imag