特牛网址导航

深度学习实战---猫狗大战(pytorch实现)_pytorch深度学习 dog and cat detection-CSDN博客

网友收藏
文章浏览阅读1.1w次,点赞8次,收藏98次。数据准备猫狗大战数据集下载链接微软的数据集已经分好类,直接使用就行,数据划分我们将猫和狗的图片分别移动到训练集和验证集中,其中90%的数据作为训练集,10%的图片作为验证集,使用shutil.move()来移动图片。新建文件夹train,test,将数据集放入train中,利用代码将10%的数据移动到test中文件移动代码import osimport shutilsource_path = r"E:\猫狗大战数据集\PetImages"train_dir = os._pytorch深度学习 dog and cat detection