本开源仓库提供了一个使用逻辑回归算法实现的猫狗图像分类可视化工具。您需要运行src/index.py
来训练模型并生成模型文件。获取模型后,通过运行src/window.py
打开可视化窗口,并选择图片进行猫狗预测。
- 数据集:请在数据集中查看包含数千张猫狗照片的预处理数据集。
- 源码
- 模型训练与生成:
src/index.py
- 可视化窗口及预测:
src/window.py
- 模型训练与生成:
-
训练模型并生成模型文件:
python src/index.py
-
打开可视化窗口并进行预测:
python src/window.py
欢迎各位开发者fork、star或提交issue,共同完善和提升这个猫狗分类可视化工具!详细英文介绍,请参阅 English README。
探索机器学习在图像分类领域的无限可能!