您当前的位置:首页 >> 视点 >  
焦点精选!基于深度学习算法U2Net网络架构的图像分割系统
来源: 哔哩哔哩      时间:2023-01-29 22:05:16


(相关资料图)

项目演示视频地址(B站): https://www.bilibili.com/video/BV1x8411G7MH/?vd_source=3464a2b24f8d1f4c170183570fefdce0

项目地址:https://mbd.pub/o/bread/Y56ZlJlq

项目代码结构截图如下: 

其中:

pre_weights:存放U2Net网络的预训练模型权重

test_pic:测试图片

predict_seg.py: 封装了调用训练好的模型进行分割的函数

tran.py: 分割模型训练脚本

train_multi_GPU.py:分割模型训练脚本(使用多GPU)

start.py: 入口程序,启动分割系统

标签: 图像分割

上一篇:

下一篇:

X 关闭

X 关闭