A simple model for Identification of nailong based on deep learning
Model reaches the accuracy of over 98%, and it can accurately tell Nailong from Konote Fujita! (of course also any other items hhh)
 Nailong |
 Konote Fujita |
Watch the demo video on bilibili: https://www.bilibili.com/video/BV18gDdY3EV6/ (CN)
Process are visualized as figure including loss and accuracy: Loss and Accuracy
Dataset are download from https://huggingface.co/datasets/XiC1/nailong-dataset
Music and inspiration come from https://www.douyin.com/video/7433341884916862260
Your computer must have a GPU
You need Python3.x, os, torch, torchvision, PIL, matplotlib, tkinter, pygame
Download the whole project, make sure your environment is complete. Run api.py.