孪生(siamese)模型训练集制作及测试模型(记录一下,大佬勿喷)
一、环境搭建
1、从github上下载源码 https://github.com/bubbliiiing/siamese-pytorch/tree/bilibili
2、根据大佬的步骤搭建环境即可(很简单,这里就不展开了)
二、创建自己的训练集
1、训练集文件格式
- image_background
- character01
- 0709_01.png
- 0709_02.png
- ……
- character02
- character03
- ……
2、使用小工具快速制作数据集
3、开始训练自己的数据集
4、模型测试