本资源为基于深度学习框架pytorch开展的水面漂浮物检测和识别研究,提供了水面垃圾数据集资源。数据集包括了数据增强制作的数据、自建的数据、网上的数据集和未开源的数据集,并提供了yolo和VOC格式的标签文件。此外,还有处理数据集和加载数据集的python文件。该资源适用于研究目标检测和分割,特别适用于水面漂浮和垃圾检测的场景,可与机器人和硬件设施配套使用,实现软件与硬件的结合。除了yolo模型,还适用于其他模型如maskrcnn、Fast和FasterRCNN等,只需自行修改相关代码和加载适应的数据集。
暂无评论