pytorch 数据集加载和处理

pytorch 数据集加载和处理

# -*- coding:utf-8 -*-
# /usr/bin/python
'''
@Author  :  Errol 
@Describe:  
@Evn     :  
@Date    :   - 
'''
import os
import torch
import pandas as pd
from skimage import io,transform
import numpy as np
import matplotlib.pyplot as plt
from torch.utils.data import Dataset,DataLoader
from torchvision import transforms,utils

import warnings
warnings.filterwarnings("ignore")
plt.ion()
landmarks_frame = pd.read_csv('data/faces/face_landmarks.csv')

n = 65
img_name = landmarks_frame.iloc[n,0]
landmarks = landmarks_frame.iloc[n,1:].as_matrix()
landmarks = landmarks.astype("float").reshape(-1,2)

print('Image name:{}'.format(img_name))
print("Landmarks shape:{}".format(landmarks.shape))
print("First 4 Landmarks:{}".format(landmarks[:4]))

def show_landmarks(image,landmarks):
    '''Show image with landmarks'