lzypython / Unet-Tom-JerryLinks

基于Unet模型和并联注意力机制实现猫和老鼠动画片的语义分割,其中并联注意力机制受CBAM注意力机制的启发,将通道注意力机制机制和空间注意力机制并联,给予不同通道不同空间不同权重。本项目使用的数据集通过爬虫获取,清洗后使用百度平台EasyData进行标注,共682张,其中545中用来训练,137张用来测试。本项目使用Visio可视化模型结构图,并对比了原版Unet、添加CBAM注意力机制和添加本项目提出的并联注意力机制三种模型。在项目的最后,通过进行模型测试,证明模型有较好的语义分割能力。
14Updated 2 months ago

Alternatives and similar repositories for Unet-Tom-Jerry

Users that are interested in Unet-Tom-Jerry are comparing it to the libraries listed below

Sorting: