dm1901的博客

我们将使用 Kaggle 提供的一个垃圾分类数据集来训练我们的模型，并使用 Flask 来构建一个简单的 Web 应用程序，允许用户上传图像并得到垃圾分类结果。我们将使用 Kaggle 提供的一个垃圾分类数据集，该数据集包含 6 种不同类型的垃圾：纸张，塑料，玻璃，金属，纺织品和垃圾堆。在这里，我们将使用 80% 的图像作为训练集，其余 20% 的图像作为测试集。我们需要确保模型的输入和输出都是正确的，并防止恶意用户上传恶意图像或攻击我们的应用程序。在训练完成后，我们需要测试我们的模型。