基于SSD的香蕉目标检测系统

这个基于SSD网络的香蕉目标检测系统使用了一个名为banana detection的数据集。你可以通过访问http://d2ldata.s3accelerate.amazonaws.com/banana_detection.zip来下载该数据集。一旦下载了数据集，你就可以运行main.py来训练网络模型ssd.pth。接着，你可以使用predictor.py来完成对图像的基本目标检测。最终结果将会在matploblib窗口中呈现，并使用框选的方式展示目标物体。如果你对计算机视觉和深度学习有兴趣，这个项目可以帮助你学习更多知识。