视觉大模型学习笔记
文章平均质量分 82
介绍视觉大模型SAM的学习笔记,主要是代码介绍和复现。后续有时间得话,可能会介绍nnUnet项目
听风吹等浪起
佛系学习,适当摆烂…
展开
-
第二章:SegmentAnyBone 分割一切骨头视觉大模型介绍
代码地址:这个分割一切骨头模型也是基于SAM的模型,官方在人体的195块骨头进行分割论文还没看,这里仅仅做复现。原创 2024-06-05 19:58:57 · 842 阅读 · 0 评论 -
MedSAM 学习笔记(续):利用训练好的权重进行gui的推理
本文将根据在自定义数据集上训练好的权重进行GUI图形化的推理因为sam模型核心思想是根据提示信息进行推理,所以推理的时候不仅仅需要图像数据,还要加上提示信息(point、bbox、txt等等)。因为无论那种提示信息输入都会很麻烦,所以源代码提供了gui的界面,直接鼠标点击绘制点或者bbox即可。原创 2024-06-04 19:04:28 · 441 阅读 · 0 评论 -
MedSAM 学习笔记(续):训练自定义数据集
下载后保存在:work_dir/SAM/sam_vit_b_01ec64.pth 目录。原创 2024-05-30 16:58:43 · 256 阅读 · 0 评论 -
第一章:MedSAM 视觉大模型介绍
这里的预处理脚本完全根据MICCAI FLARE22 挑战数据集数据集格式编写的,如果换成自己数据集的话,需要更改。其实就是目录啊、文件名之类的参数因为这个代码还挺复杂的,这里分开介绍。原创 2024-05-27 16:26:42 · 1296 阅读 · 0 评论