原始目录

 segment-anything文件夹

 

结果目录

删除了用于测试、说明的相关文件，更改了一些文件名称，使用main.py作为入口程序。

  segment-anything文件夹

具体实现 

1.删除不必要的文件

2.改segment-anything为seg

3.改amg_test.py

 

改test.py

 

改./seg/predictor.py

改./seg/build_sam.py

4.在根目录建立set_up.py 

把所有需要的.py文件放入其中。

from distutils.core import setup
from Cython.Build import cythonize
setup(ext_modules=cythonize(["amg_test.py", "test.py", "DataPrepare.py", "input_data.py", "model.py",
                             "./seg/automatic_mask_generator.py", "./seg/build_sam.py",
                             "./seg/__init__.py", "./seg/predictor.py",
                             "./seg/modeling/common.py", "./seg/modeling/image_encoder.py",
                             "./seg/modeling/__init__.py", "./seg/modeling/mask_decoder.py",
                             "./seg/modeling/sam.py", "./seg/modeling/transformer.py",
                             "./seg/modeling/prompt_encoder.py",
                             "./seg/utils/amg.py", "./seg/utils/__init__.py",
                             "./seg/utils/onnx.py", "./seg/utils/transforms.py"
                             ]))

在terminal中执行  

python set_up.py build_ext

5.删除不需要的.py文件

6.从./build/lib.linux-x86_64-cpython-310/中 复制.so文件到相应的目录

7.在根目录建立或修改main.py

from amg_test import main
from test import test
main()
test()

8.通过run  main.py可以工作了。