pytorch_小黄人项目
前提:实验使用10000个样本。 其中,下文: 1.通道处理:用于将背景影像处理为通道相同的影像。 2.查看通道分离效果:用于查看通道分离结果。 3.创建数据:创建正负样本数据集。 1. 通道处理 from PIL import Image import os """通道处理""" def convertImage ( ) : listpath = os . listdir ( "data_image" ) print ( listpath ) for index , path in enumerate ( listpath ) : img = Image . open ( os . path . join ( "data_image" , path ) ) img = img . convert ( "RGB" ) #变换通道 img = img . resize ( ( 224 , 224 ) , Image . ANTIALIAS ) #变换尺寸 # img.show() img . save ( "images/{}.jpg" . format ( index ) ) #保存 if __name__ == '__main__' : convertImage ( ) 2. 查看通道分离效果 from PIL import Image # 查看通道分离效果 img =