Skip to content

dramab/pets-recognitions

Repository files navigation

训练模型

1、使用train.ipynb文件

2、模型参数在runs\classify\train13中,每次训练会生成新的train文件其中的weights文件夹中的best.pt是最新的最好的训练参数

3、数据集存储位置:https://aistudio.baidu.com/datasetdetail/237218

使用模型预测

1、使用pre_res.py文件

2、只用更改主程序里图片地址的参数即可

3、返回图片中狗或者猫的种类的中文名称

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published