卷积神经网络(CNN)图像分类

在这个计算机视觉课业项目中，我探索了卷积神经网络(CNN)在图像分类任务中的应用，特别是在精准捕鱼的领域中的鱼类分类。我使用 PyTorch 开发了一个图像分类模型，可以根据输入图像对鱼类进行分类。这个课业项目包括两个任务：1. 训练一个模型来分类各种鱼类。2. 对训练好的模型的最后一层进行微调，以适应一些新的鱼类。训练好的模型和微调后的模型在800张测试图像上都达到了99%的准确度。