Skip to content

Latest commit

 

History

16 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 

Repository files navigation

Insect

本科科创项目:柱干昆虫检测

效果: 模型有:Resnet18、MobileNet、DenseNet、ShuffleNet、convLSTM(但是还有问题,不收敛) 在Resnet18和MobileNet上效果最好,可以达到1的准确率,在DenseNet和ShuffleNet上可以达到98%、99%的准确率

运行:

你可以在init.py文件中通过修改config类的参数来进行不同模型的训练

你需要在GetDatas.py文件中通过修改文件的路径来进行数据的读取(文件路径为音频所在的文件夹路径)

数据集所在路径:https://www.kaggle.com/datasets/potamitis/treevibes (注:由于数据集的训练集和测试集分类不均衡,所以代码中全部读取并重新切分进行训练,同时文件夹属性CONFIRMATION==NO,其中的数据不能保证标签一定正确)

文件夹说明:

由于上传文件的限制,因此数据集并没有上传(我并没有下载git)

new_model相较于model主要是resnet进行了改进: 1.加入了自注意力机制 2.将conv2d改为了conv1d

About

本科的科创项目:柱干昆虫检测

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages