基于SSD的香蕉目标检测系统
利用SSD网络进行目标的检测采用的数据集为banana detection数据集下载网页为http d2l data.s3accelerate.amazonaws.com banana detection.zip在利用数据集下载之后可以通过运行main.py来获取训练好后的网络模型ssd.pth之后再调用predictor.py即可完成对于图片的基本目标检测.最终的结果将会展示在matploblib窗口中将会用框将图片进行框选.
文件列表
SSD.rar
(预估有个15文件)
SSD
utils.py
4KB
main.py
790B
ssd_model.py
13KB
Train_fuct.py
2KB
LossFuct.py
827B
banana_src.png
156KB
banana_dataloader.py
2KB
banana_src.jpg
37KB
runing_result.jpg
12KB
暂无评论