089基于深度学习的小样本数据检测
在02中可以选择的模型有10多种,可以都训练进行对比、包括alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet和Swin transformer等10多种模型。运行01数据集文本生成制作.py可以读取图片,先进行数据增强之后,保存图片在data2文件夹下,在将每张图路径保存再txt文本中,运行02train.py可以对txt文本中的图片路径读取并训练模型,每个类别只有10张图片数据 ,
复制链接