Pytest简介
Pytest is a mature full-featured Python testing tool that helps you write better programs.The pytest framework makes it easy to write small tests, yet scales to support complex functional testing for applications and libraries.
通过官方网站介绍…
自建数据集完成二分类任务(参考文章)
1 图片预处理
1 .1 统一图片格式
找到的图片需要首先做相同尺寸的裁剪,归一化,否则会因为图片大小不同报错
RuntimeError: stack expects each tensor to be equal size,
but got [3, 667…