CNN处理图片分类

这是一个CNN大project

先看下数据集

每个文件夹有1942张图片，分别是anchor给与人安全感的人，negative坏人，positive 好人，以及测试数据集

预处理图片

DataLoader.py

import cv2
import numpy as np
import os
# 建立数据集
anchor_path = "./Data/anchor/"
pos_path= "./Data/positive/"
neg_path="./Data/negative/"
anc_imgs = [x for x in sorted(os.listdir(anchor_path)) if x[-4:] == '.png']
pos_imgs = [x for x i