【鲲码园Python】基于pytorch的鱼品种分类系统(31类)

简介

1. 编程语言: Python;

2. 数据集来自网络，其中包括13323张鱼类图片，分31类，依次为虱目鱼(Bangus)、鳙鱼(Big_Head_Carp)、黑斑鲃(Black_Spotted_Barb)、鲶鱼(Catfish)、攀鲈(Climbing_Perch)、四指丝鳍鱼(Fourfinger_Threadfin)、淡水鳗(Freshwater_Eel)、玻璃鲈鱼(Glass_Perchlet)、虾虎鱼(Goby)、金鱼(Gold_Fish)、丝足鱼(Gourami)、草鱼(Grass_Carp)、黑青斑河鲀(Green_Spotted_Puffer)、印度鲤(Indian_Carp)、太平洋海鲢(Indo-Pacific_Tarpon)、大口黑鲈(Jaguar_Gapote)、清道夫鱼(Janitor_Fish)、刀鱼(Knifefish)、长吻管鳚(Long-Snouted_Pipefish)、食蚊鱼(Mosquito_Fish)、泥鱼(Mudfish)、鲻鱼(Mullet)、巴沙鱼(Pangasius)、鲈鱼(Perch)、湄公鱼(Scat_Fish)、银刺鱼(Silver_Barb)、银鲤(Silver_Carp)、银鲈(Silver_Perch)、黑鱼(Snakehead)、海鲢(Tenpounder)、罗非鱼(Tilapia)，在项目中位于data/FishImgDataset文件夹内；

3. 深度学习使用了ResNet18(残差神经网络，属于CNN深度卷积神经网络的一种)，主要使用pytorch库进行模型构建与训练，初始阶段使用了预训练权重，测试训练轮数为20轮；

4. 图像转换、预处理使用opencv、torchvision库;

5. GUI图像界面构建展示使用PyQt5库；

内容展示

GUI界面

B站视频演示

代码结构

训练结果相关

源码+数据集+模型获取

飞书文档