EfficientSAM 项目排坑

任务

跑通这个项目代码

过程记录

创建环境

readme里没有说具体怎么配置环境，所以可能对我来说还挺困难的。
现把项目git下来：

git clone https://github.com/yformer/EfficientSAM.git
cd EfficientSAM/

看到项目里面有个setup.py，等下用他安装。但python版本还没确定，我现尝试用python3.10吧：

conda create -n EFF_SAM python=3.10
conda activate EFF_SAM
pip3 install empy rospkg pyyaml catkin_pkg
python setup.py install

没有报错，希望后面能正常使用。

运行示例

似乎预训练参数和数据都已经有了，分别在weights文件夹和figs文件夹中。看readme说有Jupyter Notebook Example，那就看看怎么用吧.
我在vscode中打开，运行给我报了些没有包的错，一个个装上就好了：

pip install ipykernel ipython
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

再运行第一步的import就每毛病了。
然后他还有个git的一步，我们已经git好了，就给他改下就行：

# !git clone https://github.com/yformer/EfficientSAM.git
import os
# os.chdir("EfficientSAM")
os.chdir('..')

下面一段程序又报错：

然后参考项目的一个issue：

糟糕！这个模块现在没有了！

没办法，只能去测试运行下另一个脚本。
报错没有cv2，那就安装：

pip install opencv-contrib-python

import没问题了，继续git SAM，然后定义函数，然后切换目录：

# !git clone https://github.com/yformer/EfficientSAM.git
import os
# os.chdir("EfficientSAM")
os.chdir('..')

然后检查模型，然后运行测试：

我的电脑在1分半跑完了。
差不多，到此为止吧。

又看了下，发现之前的问题也不是大问题。他只是squeeze_sam_model这个模型找不到了而以。把用到他的全部注释掉就能正常运行：


此外，那个单独的脚本运行起来也没毛病：

python EfficientSAM_example.py

至于训练，作者说现在代码里有很多内部依赖，没法开源，等解决了会放出来。方法类似MAE，可以把自己的数据集描述为SAM那个数据集格式然后微调。有一个人似乎在按照论文思路复现这个。