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