- 博客(4)
- 收藏
- 关注
原创 SwinUnet论文Pytorch源码复现(多分类)
SwinUnet源码链接:https://pan.baidu.com/s/1fH5jk6hkQ11W0bfY-BInFg?因为之前一次在其他博主文章下的评论,很多小伙伴跟我要SwinUnet复现源码,因为太忙(苦逼肝论文),没办法一 一回应,抱歉抱歉。数据集,我训练完成的权值文件都在文件里面。模型各项超参数为源代码中的默认值,复现出的结果比论文中低1~2两个点,可能需要去调参。可能的bug:np.bool已经被弃用,将binary.py(不在项目文件中)文件中的。numpy.bool全部替换为bool。
2024-04-02 11:45:16
1754
13
原创 直接换其他主干网络进行模型的训练
将Hiformer的models文件夹(网络模型)拖进来,还有其涉及到的一些其他文件(类)或者配置参数文件也要拖进SwinUnet的项目文件中,有utils.py(这个在SwinUnet有重名文件,换个名字就好),configs文件夹中的HiFormer_configs.pyHiFormer_configs.py,这样基本拖完了。最近想跑对比试验,但是于是直接把Hiformer的网络模型替换在SwinUnet的模型框架进行训练。我的源代码为SwinUnet的源代码,在GitHub上搜到开源代码。
2024-01-08 12:10:41
778
6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人