windows下配置mask2former(facebook版)

阳光天气發表於2024-04-11

由於此版本的mask2former官方只提供了macOS和Linux的安裝說明,所以windows安裝會趟一些坑記錄一下

1.安裝Anaconda
2.安裝PyCharm
3.建立python3.8環境(最高3.8因為有一個依賴包最高支援python3.8)
4.安裝GCC
下載地址:https://sourceforge.net/projects/mingw/
點選Download
下載完成之後,雙擊開啟點選Install,一路Next安裝

安裝GCC編譯器

開啟編譯器然後在裡點選Basic Setup,在Packpag裡找到mingw-gcc-g++ -bin,左鍵點選小框框然後點選Mark for installation

然後點選選單欄的Installation,點選Apply Changes

在彈出的對話方塊裡點選Apply,然後等待安裝完成即可

配置環境變數
先找到MinGW的安裝bin檔案的路徑,然後複製

編輯環境變數中的Path

點完之後正常情況就安裝完成了,馬上測試一下
按WIN+R鍵,然後輸入cmd,再輸入gcc -v,出現以下資訊則安裝成功

5.安裝pytorch,注意和cuda版本嚴格對應
6.安裝庫

點選檢視程式碼
pip install -U opencv-python
conda install git   
pip install fvcore  
7.安裝detectron2
點選檢視程式碼
git clone git@github.com:facebookresearch/detectron2.git
cd detectron2
pip install -e .
pip install git+https://github.com/cocodataset/panopticapi.git
pip install git+https://github.com/mcordts/cityscapesScripts.git
pip install pycocotools-windows
8.安裝detectron2的依賴庫檔案 pip install -r requirements.txt 9.編譯ops (1).把mask2former/modeling/pixel_decoder/ops 提到複製到上層資料夾 否則編譯時候報錯路徑太長生成檔案失敗 (2).報錯 ![](https://img2024.cnblogs.com/blog/3383332/202404/3383332-20240411154043699-335577736.png) 把visual studio下cl.exe的加入path環境變數 ![](https://img2024.cnblogs.com/blog/3383332/202404/3383332-20240411154355756-396242358.png) (3).報錯subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1 將 mask2former/modeling/pixel_decoder/ops

目錄下的setup.py中的“cmdclass={'build_ext': BuildExtension}”這一行改為“cmdclass={'build_ext':torch.utils.cpp_extension. BuildExtension.with_options(use_ninja=False)}”,pytorch預設使用ninjia作為backend,這裡把它禁用掉就好了
(4).在mask2former/modeling/pixel_decoder/ops下
python setup.py build install
編譯成功

相關文章