MTCNN人脸侦测项目代码-pytorch

MTCNN人脸侦测项目代码，数据是Celeba 要自己下载（也可以用其他人脸数据集，根据不同数据集调整代码）。

from torch.utils.data import Dataset
from PIL import Image
import numpy as np
from torchvision import transforms

tf = transforms.Compose（[
    transforms.ToTensor（）
]）

class MyDataset（Dataset）:

    def __init__（selfrootimg_size）:

        self.dataset = []

        #positive_file = open（f“{root}/img_size/positive.txt““r“）
        #negative_file = open（f“{root}/img_size/negative.txt““r“）
        #part_file = open（f“{root}/img_size/part.txt““r“）

        #self.dataset.extend（positive_file.readlines（））
        #self.dataset.extend（negative_file.readlines（））
        #self.dataset.extend（part_file.readlines（））

        #positive_file.close（）
        #negative_file.close（）
        #part_file.close（）

        self.img_root_dir =f“{root}/{img_size}“

        wi

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     文件        2251  2020-04-12 15:45  data.py
     文件        6033  2020-04-19 17:19  detect.py
     文件       13112  2020-04-17 17:19  gen_data.py
     文件        2294  2020-04-17 15:07  net.py
     文件          97  2020-04-18 20:30  test.py
     文件        2927  2020-04-19 21:48  train.py
     文件        1263  2020-04-18 00:43  utils.py
     文件           2  2020-04-10 23:03  MTCNN.py