0
点赞
收藏
分享

微信扫一扫

Flutter 常用插件Plugin整理并附带实例

盖码范 04-01 08:30 阅读 4

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那个数据集格式然后微调。有一个人似乎在按照论文思路复现这个。

举报

相关推荐

0 条评论