Raptor NGRANSAC兼容版

配环境

1.创建conda 环境

name: raptor
channels:
  - pytorch
  - conda-forge
dependencies:
  - python=3.7
  - pytorch=1.7.0
  - torchvision=0.8.0
  - cudatoolkit=11.0
  - pip

conda env create -f environment.yaml
conda activate raptor

2.安装ngransac

pip install opencv-python==3.4.2.17
pip install opencv-contrib-python==3.4.2.17

sudo apt-get install libopencv-dev
conda install libopencv

cd ngransac
python setup.py install

3.安装mmcv

pip install mmcv-full==1.3.17 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
pip install mmdet==2.18.0

4.Raptor安装，设定setup.py

git clone https://github.com/michalneoral/Raptor

from setuptools import setup, find_packages

print(find_packages())

setup(
    name='raptor',
    version='1.0',
    description='Raptor code by Michal Neoral',
    packages=find_packages(),
    install_requires=[
        'tqdm',
        # 'opencv-python==4.5.5.62',
        'imageio==2.13.5',
        'wandb',
        'pypng==0.0.20',
        'gdown==4.5.1',
        'kornia==0.5.3',
        'timm==0.8.2.dev0',
#        'numpy>=1.21',
#        'nptyping',
    ]#,
)

cd Raptor
pip install -v -e .

5.raft修改版安装

修改setup.py

from setuptools import setup, find_packages

print(find_packages())

setup(
    name='RAFT',
    version='0.1',
    description='',
    packages=find_packages(),
    install_requires=[
        'pypng',
        'h5py',
        'jupyter',
        'notebook',
        # 'opencv-contrib-python>=4.0.0',
        # 'opencv-python>=4.0.0',
        'Pillow',
        'scipy',
        'tqdm',
        'scikit-learn'
        ]#,

)

git clone https://github.com/michalneoral/raft_fork.git
cd raft_fork
pip install -v -e .

其他部分

模型下载、测试等
参考我的另外一篇博客：
Monocular arbitrary moving object discovery and segmentation(Raptor,NG-RANSAC 安装) 代码复现

整体库版本信息

# packages in environment at /home/ql/miniconda/envs/raptor:
#
# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                 conda_forge    conda-forge
_openmp_mutex             4.5                  2_kmp_llvm    conda-forge
addict                    2.4.0                    pypi_0    pypi
anyio                     3.7.1                    pypi_0    pypi
appdirs                   1.4.4                    pypi_0    pypi
argon2-cffi               23.1.0                   pypi_0    pypi
argon2-cffi-bindings      21.2.0                   pypi_0    pypi
attrs                     23.1.0                   pypi_0    pypi
backcall                  0.2.0                    pypi_0    pypi
beautifulsoup4            4.12.2                   pypi_0    pypi
blas                      1.0                         mkl  
bleach                    6.0.0                    pypi_0    pypi
bzip2                     1.0.8                h7b6447c_0  
ca-certificates           2023.08.22           h06a4308_0  
cairo                     1.14.12              h8948797_3  
certifi                   2023.7.22                pypi_0    pypi
cffi                      1.15.1                   pypi_0    pypi
charset-normalizer        3.3.1                    pypi_0    pypi
click                     8.1.7                    pypi_0    pypi
comm                      0.1.4                    pypi_0    pypi
cudatoolkit               11.0.3              h7761cd4_12    conda-forge
cycler                    0.11.0                   pypi_0    pypi
dataclasses               0.6                      pypi_0    pypi
debugpy                   1.7.0                    pypi_0    pypi
decorator                 5.1.1                    pypi_0    pypi
defusedxml                0.7.1                    pypi_0    pypi
docker-pycreds            0.4.0                    pypi_0    pypi
entrypoints               0.4                      pypi_0    pypi
exceptiongroup            1.1.3                    pypi_0    pypi
expat                     2.5.0                h6a678d5_0  
fastjsonschema            2.18.1                   pypi_0    pypi
ffmpeg                    4.0                  hcdf2ecd_0  
filelock                  3.12.2                   pypi_0    pypi
fontconfig                2.14.1               h52c9d5c_1  
fonttools                 4.38.0                   pypi_0    pypi
freeglut                  3.0.0                hf484d3e_5  
freetype                  2.12.1               h4a9f257_0  
fsspec                    2023.1.0                 pypi_0    pypi
future                    0.18.3                   pypi_0    pypi
gdown                     4.5.1                    pypi_0    pypi
gitdb                     4.0.11                   pypi_0    pypi
gitpython                 3.1.40                   pypi_0    pypi
glib                      2.63.1               h5a9c865_0  
graphite2                 1.3.14               h295c915_1  
h5py                      3.8.0                    pypi_0    pypi
harfbuzz                  1.8.8                hffaf4a1_0  
hdf5                      1.10.2               hba1933b_1  
huggingface-hub           0.16.4                   pypi_0    pypi
icu                       58.2                 he6710b0_3  
idna                      3.4                      pypi_0    pypi
imageio                   2.13.5                   pypi_0    pypi
importlib-metadata        6.7.0                    pypi_0    pypi
importlib-resources       5.12.0                   pypi_0    pypi
ipykernel                 6.16.2                   pypi_0    pypi
ipython                   7.34.0                   pypi_0    pypi
ipython-genutils          0.2.0                    pypi_0    pypi
ipywidgets                8.1.1                    pypi_0    pypi
jasper                    2.0.14               hd8c5072_2  
jedi                      0.19.1                   pypi_0    pypi
jinja2                    3.1.2                    pypi_0    pypi
joblib                    1.3.2                    pypi_0    pypi
jpeg                      9e                   h0b41bf4_3    conda-forge
jsonschema                4.17.3                   pypi_0    pypi
jupyter                   1.0.0                    pypi_0    pypi
jupyter-client            7.4.9                    pypi_0    pypi
jupyter-console           6.6.3                    pypi_0    pypi
jupyter-core              4.12.0                   pypi_0    pypi
jupyter-server            1.24.0                   pypi_0    pypi
jupyterlab-pygments       0.2.2                    pypi_0    pypi
jupyterlab-widgets        3.0.9                    pypi_0    pypi
kiwisolver                1.4.5                    pypi_0    pypi
kornia                    0.5.3                    pypi_0    pypi
lcms2                     2.14                 h6ed2654_0    conda-forge
ld_impl_linux-64          2.40                 h41732ed_0    conda-forge
lerc                      3.0                  h295c915_0  
libdeflate                1.8                  h7f8727e_5  
libffi                    3.2.1             hf484d3e_1007  
libgcc-ng                 13.2.0               h807b86a_2    conda-forge
libgfortran-ng            7.5.0               ha8ba4b0_17  
libgfortran4              7.5.0               ha8ba4b0_17  
libglu                    9.0.0                hf484d3e_1  
libiconv                  1.17                 h166bdaf_0    conda-forge
libnsl                    2.0.1                hd590300_0    conda-forge
libopencv                 3.4.2                hb342d67_1  
libopus                   1.3.1                h7b6447c_0  
libpng                    1.6.39               h5eee18b_0  
libstdcxx-ng              13.2.0               h7e041cc_2    conda-forge
libtiff                   4.4.0                hecacb30_2  
libuuid                   1.41.5               h5eee18b_0  
libuv                     1.46.0               hd590300_0    conda-forge
libvpx                    1.7.0                h439df22_0  
libwebp-base              1.3.2                hd590300_0    conda-forge
libxcb                    1.13              h7f98852_1004    conda-forge
libxml2                   2.9.14               h74e7548_0  
llvm-openmp               14.0.6               h9e868ea_0  
lz4-c                     1.9.4                h6a678d5_0  
markupsafe                2.1.3                    pypi_0    pypi
matplotlib                3.5.3                    pypi_0    pypi
matplotlib-inline         0.1.6                    pypi_0    pypi
mistune                   3.0.2                    pypi_0    pypi
mkl                       2021.4.0           h8d4b97c_729    conda-forge
mkl-service               2.4.0            py37h402132d_0    conda-forge
mkl_fft                   1.3.1            py37h3e078e5_1    conda-forge
mkl_random                1.2.2            py37h219a48f_0    conda-forge
mmcv-full                 1.3.17                   pypi_0    pypi
mmdet                     2.18.0                   pypi_0    pypi
nbclassic                 1.0.0                    pypi_0    pypi
nbclient                  0.7.4                    pypi_0    pypi
nbconvert                 7.6.0                    pypi_0    pypi
nbformat                  5.8.0                    pypi_0    pypi
ncurses                   6.4                  h59595ed_2    conda-forge
nest-asyncio              1.5.8                    pypi_0    pypi
ngransac                  0.0.0                    pypi_0    pypi
ninja                     1.11.1               h924138e_0    conda-forge
notebook                  6.5.6                    pypi_0    pypi
notebook-shim             0.2.3                    pypi_0    pypi
numpy                     1.21.5           py37h6c91a56_3  
numpy-base                1.21.5           py37ha15fc14_3  
olefile                   0.46                     py37_0  
opencv-contrib-python     3.4.2.17                 pypi_0    pypi
opencv-python             3.4.2.17                 pypi_0    pypi
openssl                   1.1.1w               h7f8727e_0  
packaging                 23.2                     pypi_0    pypi
pandocfilters             1.5.0                    pypi_0    pypi
parso                     0.8.3                    pypi_0    pypi
pathtools                 0.1.2                    pypi_0    pypi
pcre                      8.45                 h295c915_0  
pexpect                   4.8.0                    pypi_0    pypi
pickleshare               0.7.5                    pypi_0    pypi
pillow                    9.5.0                    pypi_0    pypi
pip                       23.3.1             pyhd8ed1ab_0    conda-forge
pixman                    0.40.0               h7f8727e_1  
pkgutil-resolve-name      1.3.10                   pypi_0    pypi
platformdirs              3.11.0                   pypi_0    pypi
prometheus-client         0.17.1                   pypi_0    pypi
prompt-toolkit            3.0.39                   pypi_0    pypi
protobuf                  4.24.4                   pypi_0    pypi
psutil                    5.9.6                    pypi_0    pypi
pthread-stubs             0.4               h36c2ea0_1001    conda-forge
ptyprocess                0.7.0                    pypi_0    pypi
pycocotools               2.0.7                    pypi_0    pypi
pycparser                 2.21                     pypi_0    pypi
pygments                  2.16.1                   pypi_0    pypi
pyparsing                 3.1.1                    pypi_0    pypi
pypng                     0.0.20                   pypi_0    pypi
pyrsistent                0.19.3                   pypi_0    pypi
pysocks                   1.7.1                    pypi_0    pypi
python                    3.7.7           hcf32534_0_cpython  
python-dateutil           2.8.2                    pypi_0    pypi
python_abi                3.7                     4_cp37m    conda-forge
pytorch                   1.7.0           py3.7_cuda11.0.221_cudnn8.0.3_0    pytorch
pyyaml                    6.0.1                    pypi_0    pypi
pyzmq                     24.0.1                   pypi_0    pypi
qtconsole                 5.4.4                    pypi_0    pypi
qtpy                      2.4.1                    pypi_0    pypi
raft                      0.1                       dev_0    <develop>
raptor                    1.0                       dev_0    <develop>
readline                  8.2                  h8228510_1    conda-forge
requests                  2.31.0                   pypi_0    pypi
scikit-learn              1.0.2                    pypi_0    pypi
scipy                     1.7.3                    pypi_0    pypi
send2trash                1.8.2                    pypi_0    pypi
sentry-sdk                1.33.1                   pypi_0    pypi
setproctitle              1.3.3                    pypi_0    pypi
setuptools                68.2.2             pyhd8ed1ab_0    conda-forge
six                       1.16.0             pyh6c4a22f_0    conda-forge
smmap                     5.0.1                    pypi_0    pypi
sniffio                   1.3.0                    pypi_0    pypi
soupsieve                 2.4.1                    pypi_0    pypi
sqlite                    3.41.2               h5eee18b_0  
tbb                       2021.8.0             hdb19cb5_0  
terminado                 0.17.1                   pypi_0    pypi
terminaltables            3.1.10                   pypi_0    pypi
threadpoolctl             3.1.0                    pypi_0    pypi
timm                      0.8.2.dev0               pypi_0    pypi
tinycss2                  1.2.1                    pypi_0    pypi
tk                        8.6.12               h1ccaba5_0  
tomli                     2.0.1                    pypi_0    pypi
torchvision               0.8.0                py37_cu110    pytorch
tornado                   6.2                      pypi_0    pypi
tqdm                      4.66.1                   pypi_0    pypi
traitlets                 5.9.0                    pypi_0    pypi
typing_extensions         4.7.1              pyha770c72_0    conda-forge
urllib3                   2.0.7                    pypi_0    pypi
wandb                     0.15.12                  pypi_0    pypi
wcwidth                   0.2.9                    pypi_0    pypi
webencodings              0.5.1                    pypi_0    pypi
websocket-client          1.6.1                    pypi_0    pypi
wheel                     0.41.3             pyhd8ed1ab_0    conda-forge
widgetsnbextension        4.0.9                    pypi_0    pypi
xorg-libxau               1.0.11               hd590300_0    conda-forge
xorg-libxdmcp             1.1.3                h7f98852_0    conda-forge
xz                        5.2.6                h166bdaf_0    conda-forge
yapf                      0.40.2                   pypi_0    pypi
zipp                      3.15.0                   pypi_0    pypi
zlib                      1.2.13               h5eee18b_0  
zstd                      1.5.2                ha4553b6_0