Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

artist_tagger

  • 基于resnet50,进行多标签分类的训练示例和运行脚本,可用于训练画师串反推模型
  • train用于训练,run用于运行已有模型
  • 代码中默认使用的csv编码是latin1,也就是说csv文件中不能出现中文,如有需求可自行更改
  • 训练时train.csv填图片的名称(包括后缀)和对应的tags,注意tag之间用英文逗号分割,不能带有权重语法,所有可用于标注的tag见tags.csv
  • 如果图片的后缀不是jpg,jepg,png则统一在train.csv里填后缀为jpg,因为别的格式(包括GIF),会被处理成jpg
  • gif取的是第一帧
  • kaggle直达链接
  • https://www.kaggle.com/code/spawnerqwq/train-tagger
  • https://www.kaggle.com/code/spawnerqwq/run-tagger

About

基于resnet50,进行多标签分类的训练示例和运行脚本,可用于训练画师串反推模型

Resources

Stars

7 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages