SAM2 分割一切 环境配置及运行测试的主要问题

1.环境配置

按照REDEME的流程，已安装python3.11，pytorch2.4.0+cu124

cd到segment-anything-2-main文件夹，执行：

pip install -e .

 发生以下错误：

解决办法：在 setup.py的第21行，加入encoding='utf-8'

with open("README.md", "r", encoding='utf-8') as f:
    LONG_DESCRIPTION = f.read()

再次执行pip install -e .   即可成功安装。

 

 2.模型下载

 打开SAM2的GitHub官方地址，把下面4个模型全部下载，移动到segment-anything-2-main/cheakpoints目录下。

 

 

3.代码测试 

 安装matplotlib、opencv-python、jupyter等包，哪个包爆红安装哪个。运行image_predictor_example.ipynb

 

 全部运行时报错：1Torch was not compiled with flash attention.

 解决方案：在sam2/modeling/sam/transformer.py中禁用flash attention即可。在24行添加以下代码：

 

 问题解决，代码跑通。