mzdluo123 / mzdluo123.github.io

博客
5 stars 2 forks source link

充分使用Colab的免费TPU | RainChan的小博客 #39

Open mzdluo123 opened 2 years ago

mzdluo123 commented 2 years ago

https://rainchan.win/2022/02/07/%E5%85%85%E5%88%86%E4%BD%BF%E7%94%A8Colab%E7%9A%84%E5%85%8D%E8%B4%B9TPU/

前言好久不见!这几天虽然放假在家,但是各种事情还是非常多,忙里偷闲做了一个给动漫图片添加tag的神经网络模型。因此写下这篇文章来记录一下过程。 给动漫图片添加tag属于多标签分类问题:一张图片可以有多个标签。 这类问题不能简单使用准确率来判断模型的性能,因此我用通过召回率和精确率进行计算的F1值来评价模型的性能。 最终结果:模型在 danbooru网站上的20万张图片上进行训练,能够在470个ta