1.可以自定义一个新的数据源 参考: internal class VirtualDataSource : DataSourceBase { private int FVirtualRowsCount; public int VirtualRowsCount { get { return FVirtualRowsCount; } set { FVirtualRowsCount = value; } } #region Protected Me…
1.将图片的路径和标签写入csv文件并实现读取 # 创建一个文件,包含image,存放方式:label pokemeon\\mew\\0001.jpg,0 def load_csv(self,filename): if not os.path.exists(os.path.join(self.root,filename)): images = [] # 将所有的信息组成一个列表,类别信息通过中间的一个路径判断 for name in self.name2label.keys(): # pokem…
对于自定义数据集的图片任务,通用流程一般分为以下几个步骤: Load data Train-Val-Test Build model Transfer Learning 其中大部分精力会花在数据的准备和预处理上,本文用一种较为通用的数据处理手段,并通过手动构建,简单模型, 层数较深的resnet网络,和基于VGG19的迁移学习. 你可以通过这个例子,快速搭建网络,并训练处一个较为满意的结果. 1. Load data 数据集来自Pokemon的5分类数据, 每一种的图片数量为200多张,是一个较…