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