(ocinet_py38) root@davinci-mini:~/OCINet-main# conda deactivate root@davinci-mini:~/OCINet-main# conda create -n ocinet_py38 python=3.8 -y conda activate ocinet_py38 Retrieving notices: ...working... done Collecting package metadata (current_repodata.json): done Solving environment: done ==> WARNING: A newer version of conda exists. <== current version: 23.5.0 latest version: 25.11.1 Please update conda by running $ conda update -n base -c defaults conda Or to minimize the number of packages updated during conda update use conda install conda=25.11.1 ## Package Plan ## environment location: /usr/local/miniconda3/envs/ocinet_py38 added / updated specs: - python=3.8 The following NEW packages will be INSTALLED: _libgcc_mutex pkgs/main/linux-aarch64::_libgcc_mutex-0.1-main _openmp_mutex pkgs/main/linux-aarch64::_openmp_mutex-5.1-51_gnu ca-certificates pkgs/main/linux-aarch64::ca-certificates-2025.12.2-hd43f75c_0 ld_impl_linux-aar~ pkgs/main/linux-aarch64::ld_impl_linux-aarch64-2.44-h4e4ac68_2 libffi pkgs/main/linux-aarch64::libffi-3.4.4-h419075a_1 libgcc pkgs/main/linux-aarch64::libgcc-15.2.0-hc18542e_7 libgcc-ng pkgs/main/linux-aarch64::libgcc-ng-15.2.0-h51576c1_7 libgomp pkgs/main/linux-aarch64::libgomp-15.2.0-hf47c802_7 libstdcxx pkgs/main/linux-aarch64::libstdcxx-15.2.0-h9538471_7 libstdcxx-ng pkgs/main/linux-aarch64::libstdcxx-ng-15.2.0-h719ae8e_7 libxcb pkgs/main/linux-aarch64::libxcb-1.17.0-hf66535e_0 libzlib pkgs/main/linux-aarch64::libzlib-1.3.1-h998d150_0 ncurses pkgs/main/linux-aarch64::ncurses-6.5-h419075a_0 openssl pkgs/main/linux-aarch64::openssl-3.0.18-h8a9c3cf_0 pip conda-forge/noarch::pip-22.1.2-pyhd8ed1ab_0 pthread-stubs pkgs/main/linux-aarch64::pthread-stubs-0.3-hfd63f10_1 python pkgs/main/linux-aarch64::python-3.8.20-h4bb2201_0 readline pkgs/main/linux-aarch64::readline-8.3-h886d1d0_0 setuptools pkgs/main/linux-aarch64::setuptools-72.1.0-py38hd43f75c_0 sqlite pkgs/main/linux-aarch64::sqlite-3.51.0-hc5b20ee_0 tk pkgs/main/linux-aarch64::tk-8.6.15-h987d8db_0 wheel pkgs/main/noarch::wheel-0.37.1-pyhd3eb1b0_0 xorg-libx11 pkgs/main/linux-aarch64::xorg-libx11-1.8.12-hf66535e_1 xorg-libxau pkgs/main/linux-aarch64::xorg-libxau-1.0.12-hf66535e_0 xorg-libxdmcp pkgs/main/linux-aarch64::xorg-libxdmcp-1.1.5-hf66535e_0 xorg-xorgproto pkgs/main/linux-aarch64::xorg-xorgproto-2024.1-h998d150_1 xz pkgs/main/linux-aarch64::xz-5.6.4-h998d150_1 zlib pkgs/main/linux-aarch64::zlib-1.3.1-h998d150_0 Downloading and Extracting Packages Preparing transaction: done Verifying transaction: done Executing transaction: done # # To activate this environment, use # # $ conda activate ocinet_py38 # # To deactivate an active environment, use # # $ conda deactivate (ocinet_py38) root@davinci-mini:~/OCINet-main# pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu Looking in indexes: https://pypi.doubanio.com/simple, https://download.pytorch.org/whl/cpu Collecting torch Using cached https://download.pytorch.org/whl/cpu/torch-2.4.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (89.7 MB) Collecting torchvision Downloading https://mirrors.cloud.tencent.com/pypi/packages/95/14/b7dadde3ef929936e2139aa3f51f078887b6cc8bee702410979b929224e8/torchvision-0.19.1-cp38-cp38-manylinux2014_aarch64.whl (14.1 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 14.1/14.1 MB 5.5 MB/s eta 0:00:00 Collecting filelock Using cached filelock-3.16.1-py3-none-any.whl (16 kB) Collecting jinja2 Using cached https://download.pytorch.org/whl/jinja2-3.1.6-py3-none-any.whl (134 kB) Collecting typing-extensions>=4.8.0 Using cached https://download.pytorch.org/whl/typing_extensions-4.15.0-py3-none-any.whl (44 kB) Collecting sympy Using cached sympy-1.13.3-py3-none-any.whl (6.2 MB) Collecting networkx Using cached networkx-3.1-py3-none-any.whl (2.1 MB) Collecting fsspec Using cached fsspec-2025.3.0-py3-none-any.whl (193 kB) Collecting pillow!=8.3.*,>=5.3.0 Using cached pillow-10.4.0-cp38-cp38-manylinux_2_28_aarch64.whl (4.4 MB) Collecting numpy Using cached numpy-1.24.4-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (14.0 MB) Collecting typing-extensions>=4.8.0 Using cached https://download.pytorch.org/whl/typing_extensions-4.14.0-py3-none-any.whl (43 kB) Using cached https://mirrors.cloud.tencent.com/pypi/packages/8b/54/b1ae86c0973cc6f0210b53d508ca3641fb6d0c56823f288d108bc7ab3cc8/typing_extensions-4.13.2-py3-none-any.whl (45 kB) Collecting MarkupSafe>=2.0 Using cached https://download.pytorch.org/whl/MarkupSafe-2.1.5-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (26 kB) Collecting mpmath<1.4,>=1.1.0 Using cached mpmath-1.3.0-py3-none-any.whl (536 kB) Installing collected packages: mpmath, typing-extensions, sympy, pillow, numpy, networkx, MarkupSafe, fsspec, filelock, jinja2, torch, torchvision ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. te 0.4.0 requires attrs, which is not installed. te 0.4.0 requires cloudpickle, which is not installed. te 0.4.0 requires decorator, which is not installed. te 0.4.0 requires psutil, which is not installed. te 0.4.0 requires scipy, which is not installed. te 0.4.0 requires synr==0.5.0, which is not installed. te 0.4.0 requires tornado, which is not installed. schedule-search 0.0.1 requires absl-py, which is not installed. schedule-search 0.0.1 requires decorator, which is not installed. opc-tool 0.1.0 requires attrs, which is not installed. opc-tool 0.1.0 requires decorator, which is not installed. opc-tool 0.1.0 requires psutil, which is not installed. auto-tune 0.1.0 requires decorator, which is not installed. Successfully installed MarkupSafe-2.1.5 filelock-3.16.1 fsspec-2025.3.0 jinja2-3.1.6 mpmath-1.3.0 networkx-3.1 numpy-1.24.4 pillow-10.4.0 sympy-1.13.3 torch-2.4.1 torchvision-0.19.1 typing-extensions-4.13.2 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv (ocinet_py38) root@davinci-mini:~/OCINet-main# conda install -c conda-forge opencv scikit-image matplotlib tqdm scipy Collecting package metadata (current_repodata.json): - failed CondaError: KeyboardInterrupt (ocinet_py38) root@davinci-mini:~/OCINet-main# pip install opencv-python scikit-image matplotlib tqdm scipy pandas seaborn Looking in indexes: https://pypi.doubanio.com/simple Collecting opencv-python Using cached https://mirrors.cloud.tencent.com/pypi/packages/62/3a/440bd64736cf8116f01f3b7f9f2e111afb2e02beb2ccc08a6458114a6b5d/opencv_python-4.12.0.88-cp37-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.whl (45.9 MB) Collecting scikit-image Downloading https://mirrors.cloud.tencent.com/pypi/packages/fa/2b/ffecc6f29b48d1d46dc3bb7b4c908490260c3a0d69ac2d248d846b90d505/scikit_image-0.21.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (13.3 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 13.3/13.3 MB 4.9 MB/s eta 0:00:00 Collecting matplotlib Downloading https://mirrors.cloud.tencent.com/pypi/packages/9b/9d/8bd37c86b79312c9dbcfa379dec32303f9b38e8456e0829d7e666a0e0a05/matplotlib-3.7.5-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (11.4 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 11.4/11.4 MB 5.1 MB/s eta 0:00:00 Collecting tqdm Downloading https://mirrors.cloud.tencent.com/pypi/packages/d0/30/dc54f88dd4a2b5dc8a0279bdd7270e735851848b762aeb1c1184ed1f6b14/tqdm-4.67.1-py3-none-any.whl (78 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 78.5/78.5 kB 1.7 MB/s eta 0:00:00 Collecting scipy Downloading https://mirrors.cloud.tencent.com/pypi/packages/d2/b5/ff61b79ad0ebd15d87ade10e0f4e80114dd89fac34a5efade39e99048c91/scipy-1.10.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (31.0 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 31.0/31.0 MB 3.5 MB/s eta 0:00:00 Collecting pandas Downloading https://mirrors.cloud.tencent.com/pypi/packages/a7/87/828d50c81ce0f434163bf70b925a0eec6076808e0bca312a79322b141f66/pandas-2.0.3-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (11.7 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 11.7/11.7 MB 5.2 MB/s eta 0:00:00 Collecting seaborn Using cached https://mirrors.cloud.tencent.com/pypi/packages/83/11/00d3c3dfc25ad54e731d91449895a79e4bf2384dc3ac01809010ba88f6d5/seaborn-0.13.2-py3-none-any.whl (294 kB) Requirement already satisfied: numpy<2.0 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from opencv-python) (1.24.4) Collecting imageio>=2.27 Downloading https://mirrors.cloud.tencent.com/pypi/packages/1e/b7/02adac4e42a691008b5cfb31db98c190e1fc348d1521b9be4429f9454ed1/imageio-2.35.1-py3-none-any.whl (315 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 315.4/315.4 kB 2.8 MB/s eta 0:00:00 Requirement already satisfied: pillow>=9.0.1 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from scikit-image) (10.4.0) Collecting PyWavelets>=1.1.1 Downloading https://mirrors.cloud.tencent.com/pypi/packages/cd/c1/132756d0033b37f4013299ac048bf34d5094673712984edb9e90e8d8a179/PyWavelets-1.4.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (6.9 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 6.9/6.9 MB 5.4 MB/s eta 0:00:00 Collecting packaging>=21 Using cached https://mirrors.cloud.tencent.com/pypi/packages/20/12/38679034af332785aac8774540895e234f4d07f7545804097de4b666afd8/packaging-25.0-py3-none-any.whl (66 kB) Collecting tifffile>=2022.8.12 Downloading https://mirrors.cloud.tencent.com/pypi/packages/06/a3/68d17088a4f09565bc7341fd20490da8191ec4cddde479daaabbe07bb603/tifffile-2023.7.10-py3-none-any.whl (220 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 220.9/220.9 kB 2.5 MB/s eta 0:00:00 Collecting lazy_loader>=0.2 Downloading https://mirrors.cloud.tencent.com/pypi/packages/83/60/d497a310bde3f01cb805196ac61b7ad6dc5dcf8dce66634dc34364b20b4f/lazy_loader-0.4-py3-none-any.whl (12 kB) Requirement already satisfied: networkx>=2.8 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from scikit-image) (3.1) Collecting python-dateutil>=2.7 Using cached https://mirrors.cloud.tencent.com/pypi/packages/ec/57/56b9bcc3c9c6a792fcbaf139543cee77261f3651ca9da0c93f5c1221264b/python_dateutil-2.9.0.post0-py2.py3-none-any.whl (229 kB) Collecting contourpy>=1.0.1 Downloading https://mirrors.cloud.tencent.com/pypi/packages/9b/cb/08f884c4c2efd433a38876b1b8069bfecef3f2d21ff0ce635d455962f70f/contourpy-1.1.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (285 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 285.8/285.8 kB 2.0 MB/s eta 0:00:00 Collecting fonttools>=4.22.0 Downloading https://mirrors.cloud.tencent.com/pypi/packages/17/b2/4e887b674938b4c3848029a4134ac90dd8653ea80b4f464fa1edeae37f25/fonttools-4.57.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (4.6 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.6/4.6 MB 5.4 MB/s eta 0:00:00 Collecting pyparsing>=2.3.1 Using cached https://mirrors.cloud.tencent.com/pypi/packages/e5/0c/0e3c05b1c87bb6a1c76d281b0f35e78d2d80ac91b5f8f524cebf77f51049/pyparsing-3.1.4-py3-none-any.whl (104 kB) Collecting importlib-resources>=3.2.0 Using cached https://mirrors.cloud.tencent.com/pypi/packages/e1/6a/4604f9ae2fa62ef47b9de2fa5ad599589d28c9fd1d335f32759813dfa91e/importlib_resources-6.4.5-py3-none-any.whl (36 kB) Collecting kiwisolver>=1.0.1 Downloading https://mirrors.cloud.tencent.com/pypi/packages/54/a4/3b5a2542429e182a4df0528214e76803f79d016110f5e67c414a0357cd7d/kiwisolver-1.4.7-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.4 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.4/1.4 MB 2.0 MB/s eta 0:00:00 Collecting cycler>=0.10 Using cached https://mirrors.cloud.tencent.com/pypi/packages/e7/05/c19819d5e3d95294a6f5947fb9b9629efb316b96de511b418c53d245aae6/cycler-0.12.1-py3-none-any.whl (8.3 kB) Collecting tzdata>=2022.1 Downloading https://mirrors.cloud.tencent.com/pypi/packages/c7/b0/003792df09decd6849a5e39c28b513c06e84436a54440380862b5aeff25d/tzdata-2025.3-py2.py3-none-any.whl (348 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 348.5/348.5 kB 2.4 MB/s eta 0:00:00 Collecting pytz>=2020.1 Using cached https://mirrors.cloud.tencent.com/pypi/packages/81/c4/34e93fe5f5429d7570ec1fa436f1986fb1f00c3e0f43a589fe2bbcd22c3f/pytz-2025.2-py2.py3-none-any.whl (509 kB) Collecting zipp>=3.1.0 Using cached https://mirrors.cloud.tencent.com/pypi/packages/62/8b/5ba542fa83c90e09eac972fc9baca7a88e7e7ca4b221a89251954019308b/zipp-3.20.2-py3-none-any.whl (9.2 kB) Collecting six>=1.5 Using cached https://mirrors.cloud.tencent.com/pypi/packages/b7/ce/149a00dd41f10bc29e5921b496af8b574d8413afcd5e30dfa0ed46c2cc5e/six-1.17.0-py2.py3-none-any.whl (11 kB) Installing collected packages: pytz, zipp, tzdata, tqdm, tifffile, six, scipy, PyWavelets, pyparsing, packaging, opencv-python, kiwisolver, imageio, fonttools, cycler, contourpy, python-dateutil, lazy_loader, importlib-resources, scikit-image, pandas, matplotlib, seaborn ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts. te 0.4.0 requires attrs, which is not installed. te 0.4.0 requires cloudpickle, which is not installed. te 0.4.0 requires decorator, which is not installed. te 0.4.0 requires psutil, which is not installed. te 0.4.0 requires synr==0.5.0, which is not installed. te 0.4.0 requires tornado, which is not installed. auto-tune 0.1.0 requires decorator, which is not installed. Successfully installed PyWavelets-1.4.1 contourpy-1.1.1 cycler-0.12.1 fonttools-4.57.0 imageio-2.35.1 importlib-resources-6.4.5 kiwisolver-1.4.7 lazy_loader-0.4 matplotlib-3.7.5 opencv-python-4.12.0.88 packaging-25.0 pandas-2.0.3 pyparsing-3.1.4 python-dateutil-2.9.0.post0 pytz-2025.2 scikit-image-0.21.0 scipy-1.10.1 seaborn-0.13.2 six-1.17.0 tifffile-2023.7.10 tqdm-4.67.1 tzdata-2025.3 zipp-3.20.2 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv (ocinet_py38) root@davinci-mini:~/OCINet-main# python test_OCINet_30_images.py Traceback (most recent call last): File "test_OCINet_30_images.py", line 11, in <module> from model.OCINet_models import OCINet File "/root/OCINet-main/model/OCINet_models.py", line 6, in <module> from model.pvtv2 import pvt_v2_b2 File "/root/OCINet-main/model/pvtv2.py", line 6, in <module> from timm.models.layers import DropPath, to_2tuple, trunc_normal_ ModuleNotFoundError: No module named 'timm' (ocinet_py38) root@davinci-mini:~/OCINet-main# pip install timm Looking in indexes: https://pypi.doubanio.com/simple Collecting timm Downloading https://mirrors.cloud.tencent.com/pypi/packages/d6/14/fc04d491527b774ec7479897f5861959209de1480e4c4cd32ed098ff8bea/timm-1.0.22-py3-none-any.whl (2.5 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.5/2.5 MB 5.1 MB/s eta 0:00:00 Requirement already satisfied: torch in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from timm) (2.4.1) Collecting pyyaml Downloading https://mirrors.cloud.tencent.com/pypi/packages/02/72/d972384252432d57f248767556ac083793292a4adf4e2d85dfe785ec2659/PyYAML-6.0.3-cp38-cp38-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl (795 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 795.1/795.1 kB 2.2 MB/s eta 0:00:00 Collecting huggingface_hub Using cached https://mirrors.cloud.tencent.com/pypi/packages/cb/bd/1a875e0d592d447cbc02805fd3fe0f497714d6a2583f59d14fa9ebad96eb/huggingface_hub-0.36.0-py3-none-any.whl (566 kB) Collecting safetensors Downloading https://mirrors.cloud.tencent.com/pypi/packages/5d/9a/add3e6fef267658075c5a41573c26d42d80c935cdc992384dfae435feaef/safetensors-0.5.3-cp38-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (459 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 459.5/459.5 kB 3.2 MB/s eta 0:00:00 Requirement already satisfied: torchvision in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from timm) (0.19.1) Collecting requests Using cached https://mirrors.cloud.tencent.com/pypi/packages/7c/e4/56027c4a6b4ae70ca9de302488c5ca95ad4a39e190093d6c1a8ace08341b/requests-2.32.4-py3-none-any.whl (64 kB) Requirement already satisfied: tqdm>=4.42.1 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from huggingface_hub->timm) (4.67.1) Requirement already satisfied: packaging>=20.9 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from huggingface_hub->timm) (25.0) Requirement already satisfied: fsspec>=2023.5.0 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from huggingface_hub->timm) (2025.3.0) Requirement already satisfied: filelock in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from huggingface_hub->timm) (3.16.1) Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from huggingface_hub->timm) (4.13.2) Collecting hf-xet<2.0.0,>=1.1.3 Using cached https://mirrors.cloud.tencent.com/pypi/packages/46/92/3f7ec4a1b6a65bf45b059b6d4a5d38988f63e193056de2f420137e3c3244/hf_xet-1.2.0-cp37-abi3-manylinux_2_28_aarch64.whl (3.2 MB) Requirement already satisfied: jinja2 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from torch->timm) (3.1.6) Requirement already satisfied: networkx in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from torch->timm) (3.1) Requirement already satisfied: sympy in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from torch->timm) (1.13.3) Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from torchvision->timm) (10.4.0) Requirement already satisfied: numpy in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from torchvision->timm) (1.24.4) Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from jinja2->torch->timm) (2.1.5) Collecting charset_normalizer<4,>=2 Downloading https://mirrors.cloud.tencent.com/pypi/packages/ec/7c/b92d1d1dcffc34592e71ea19c882b6709e43d20fa498042dea8b815638d7/charset_normalizer-3.4.4-cp38-cp38-manylinux2014_aarch64.manylinux_2_17_aarch64.manylinux_2_28_aarch64.whl (143 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 143.1/143.1 kB 4.1 MB/s eta 0:00:00 Collecting idna<4,>=2.5 Using cached https://mirrors.cloud.tencent.com/pypi/packages/0e/61/66938bbb5fc52dbdf84594873d5b51fb1f7c7794e9c0f5bd885f30bc507b/idna-3.11-py3-none-any.whl (71 kB) Collecting certifi>=2017.4.17 Using cached https://mirrors.cloud.tencent.com/pypi/packages/e6/ad/3cc14f097111b4de0040c83a525973216457bbeeb63739ef1ed275c1c021/certifi-2026.1.4-py3-none-any.whl (152 kB) Collecting urllib3<3,>=1.21.1 Using cached https://mirrors.cloud.tencent.com/pypi/packages/ce/d9/5f4c13cecde62396b0d3fe530a50ccea91e7dfc1ccf0e09c228841bb5ba8/urllib3-2.2.3-py3-none-any.whl (126 kB) Requirement already satisfied: mpmath<1.4,>=1.1.0 in /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages (from sympy->torch->timm) (1.3.0) Installing collected packages: urllib3, safetensors, pyyaml, idna, hf-xet, charset_normalizer, certifi, requests, huggingface_hub, timm Successfully installed certifi-2026.1.4 charset_normalizer-3.4.4 hf-xet-1.2.0 huggingface_hub-0.36.0 idna-3.11 pyyaml-6.0.3 requests-2.32.4 safetensors-0.5.3 timm-1.0.22 urllib3-2.2.3 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv (ocinet_py38) root@davinci-mini:~/OCINet-main# python test_OCINet_30_images.py /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages/timm/models/layers/__init__.py:48: FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.layers", FutureWarning) /usr/local/miniconda3/envs/ocinet_py38/lib/python3.8/site-packages/timm/models/registry.py:4: FutureWarning: Importing from timm.models.registry is deprecated, please import via timm.models warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.models", FutureWarning) /root/OCINet-main/model/pvtv2.py:388: UserWarning: Overwriting pvt_v2_b0 in registry with model.pvtv2.pvt_v2_b0. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b0(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:398: UserWarning: Overwriting pvt_v2_b1 in registry with model.pvtv2.pvt_v2_b1. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b1(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:406: UserWarning: Overwriting pvt_v2_b2 in registry with model.pvtv2.pvt_v2_b2. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b2(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:414: UserWarning: Overwriting pvt_v2_b3 in registry with model.pvtv2.pvt_v2_b3. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b3(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:422: UserWarning: Overwriting pvt_v2_b4 in registry with model.pvtv2.pvt_v2_b4. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b4(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:431: UserWarning: Overwriting pvt_v2_b5 in registry with model.pvtv2.pvt_v2_b5. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b5(PyramidVisionTransformerImpr): 正在加载模型... /root/OCINet-main/model/OCINet_models.py:393: FutureWarning: You are using torch.load with weights_only=False (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for weights_only will be flipped to True. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via torch.serialization.add_safe_globals. We recommend you start setting weights_only=True for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature. save_model = torch.load(path) test_OCINet_30_images.py:35: FutureWarning: You are using torch.load with weights_only=False (the current default value), which uses the default pickle module implicitly. It is possible to construct malicious pickle data which will execute arbitrary code during unpickling (See https://github.com/pytorch/pytorch/blob/main/SECURITY.md#untrusted-models for more details). In a future release, the default value for weights_only will be flipped to True. This limits the functions that could be executed during unpickling. Arbitrary objects will no longer be allowed to be loaded via this mode unless they are explicitly allowlisted by the user via torch.serialization.add_safe_globals. We recommend you start setting weights_only=True for any use case where you don't have full control of the loaded file. Please open an issue on GitHub for any issues related to this experimental feature. checkpoint = torch.load(model_path, map_location='cpu') 成功加载模型: ./models/OCINet.pth 模型准备就绪 ============================================================ 限制设置: 每个数据集最多处理 30 张图像 ============================================================ ================================================== 开始处理数据集: 965 图像路径: ./dataset/Rail/1130/965/img/ 保存路径: ./models/965/ ================================================== 发现 965 张图像，将处理前 30 张 预计总时间: 90.0秒 (1.5分钟) 处理 965: 0%| | 0/30 [00:05<?, ?图像/s, 当前图像=s0000510b_1_0.png, 推理时间=5.32s, 平均时间=5.32s, 剩余时间=154s] [1/30] 图像: s0000510b_1_0.png | 本次: 5.32s | 平均: 5.32s | 剩余: 154.4s 处理 965: 13%|████████ | 4/30 [00:24<02:08, 4.93s/图像, 当前图像=s0000510b_1_4.png, 推理时间=4.76s, 平均时间=4.88s, 剩余时间=122s] [5/30] 图像: s0000510b_1_4.png | 本次: 4.76s | 平均: 4.88s | 剩余: 122.1s 处理 965: 30%|██████████████████▎ | 9/30 [00:48<01:40, 4.79s/图像, 当前图像=s0000510b_2_4.png, 推理时间=4.80s, 平均时间=4.79s, 剩余时间=96s] [10/30] 图像: s0000510b_2_4.png | 本次: 4.80s | 平均: 4.79s | 剩余: 95.8s 处理 965: 47%|████████████████████████████ | 14/30 [01:12<01:14, 4.67s/图像, 当前图像=s0000511b_0_4.png, 推理时间=4.67s, 平均时间=4.71s, 剩余时间=71s] [15/30] 图像: s0000511b_0_4.png | 本次: 4.67s | 平均: 4.71s | 剩余: 70.7s 处理 965: 63%|██████████████████████████████████████ | 19/30 [01:35<00:51, 4.71s/图像, 当前图像=s0000513b_0_0.png, 推理时间=4.51s, 平均时间=4.69s, 剩余时间=47s] [20/30] 图像: s0000513b_0_0.png | 本次: 4.51s | 平均: 4.69s | 剩余: 46.9s 处理 965: 80%|████████████████████████████████████████████████ | 24/30 [01:59<00:28, 4.69s/图像, 当前图像=s0000513b_1_0.png, 推理时间=4.77s, 平均时间=4.68s, 剩余时间=23s] [25/30] 图像: s0000513b_1_0.png | 本次: 4.77s | 平均: 4.68s | 剩余: 23.4s 处理 965: 97%|██████████████████████████████████████████████████████████▉ | 29/30 [02:22<00:04, 4.75s/图像, 当前图像=s0000513b_2_0.png, 推理时间=4.66s, 平均时间=4.68s, 剩余时间=0s] [30/30] 图像: s0000513b_2_0.png | 本次: 4.66s | 平均: 4.68s | 剩余: 0.0s 处理 965: 100%|█████████████████████████████████████████████████████████████| 30/30 [02:22<00:00, 4.76s/图像, 当前图像=s0000513b_2_0.png, 推理时间=4.66s, 平均时间=4.68s, 剩余时间=0s] 数据集 965 处理完成! 实际处理图像数: 30/30 总时间: 140.35秒 平均每张图像推理时间: 4.678秒 FPS: 0.21 提示: 完整数据集(965张)预计需要: 4514.7秒 (75.2分钟) ==================================================

(ocinet_npu) root@davinci-mini:~/OCINet-main# pip install onnx==1.14.1 -i https://pypi.doubanio.com/simple Looking in indexes: https://pypi.doubanio.com/simple Collecting onnx==1.14.1 Downloading https://mirrors.cloud.tencent.com/pypi/packages/13/b8/77f39777e48046672efc747cb652f762828a0b7d3fd226be160e65a01d5b/onnx-1.14.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (15.0 MB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 15.0/15.0 MB 2.0 MB/s eta 0:00:00 Collecting protobuf>=3.20.2 Downloading https://mirrors.cloud.tencent.com/pypi/packages/6c/04/98f6f8cf5b07ab1294c13f34b4e69b3722bb609c5b701d6c169828f9f8aa/protobuf-5.29.5-cp38-abi3-manylinux2014_aarch64.whl (319 kB) ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 319.8/319.8 kB 1.1 MB/s eta 0:00:00 Requirement already satisfied: typing-extensions>=3.6.2.1 in /usr/local/miniconda3/envs/ocinet_npu/lib/python3.8/site-packages (from onnx==1.14.1) (4.13.2) Requirement already satisfied: numpy in /usr/local/miniconda3/envs/ocinet_npu/lib/python3.8/site-packages (from onnx==1.14.1) (1.24.4) Installing collected packages: protobuf, onnx Successfully installed onnx-1.14.1 protobuf-5.29.5 WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv (ocinet_npu) root@davinci-mini:~/OCINet-main# cat > export_ocinet_onnx.py <<'PY' import os import torch from model.OCINet_models import OCINet def remap_key(k: str) -> str: # 1) 去掉 module. if k.startswith("module."): k = k[len("module."):] # 2) 权重里用 e.，代码里用 backbone. if k.startswith("e."): k = "backbone." + k[len("e."):] # 3) Normalization -> ChannelNormalization if k.startswith("Normalization_"): k = "ChannelNormalization_" + k[len("Normalization_"):] # 4) SelfAtt -> SpatialSelfAtt，但避免二次替换 # 先把错误的 SpatialSpatialSelfAtt 折叠掉（保险） k = k.replace("SpatialSpatialSelfAtt", "SpatialSelfAtt") # 再只对还没带 SpatialSelfAtt 的情况替换 if "SpatialSelfAtt" not in k: k = k.replace("SelfAtt", "SpatialSelfAtt") return k model = OCINet() ckpt = torch.load("./models/OCINet.pth", map_location="cpu") state_dict = ckpt.get("state_dict", ckpt) if isinstance(ckpt, dict) else ckpt new_state = {} drop_cnt = 0 for k, v in state_dict.items(): if k.endswith("num_batches_tracked"): drop_cnt += 1 continue new_state[remap_key(k)] = v missing, unexpected = model.load_state_dict(new_state, strict=False) model.eval() print("Dropped keys (num_batches_tracked):", drop_cnt) print("Missing keys:", len(missing)) print("Unexpected keys:", len(unexpected)) print("Missing sample:", missing[:10]) print("Unexpected sample:", unexpected[:10]) dummy = torch.randn(1, 3, 352, 352) out_path = "ocinet_352.onnx" torch.onnx.export( model, dummy, ls -lh ./ocinet_352.onnxx.pyath(out_path))s_sig","s2_sig","s3_sig","s4_sig"], /usr/local/miniconda3/envs/ocinet_npu/lib/python3.8/site-packages/timm/models/layers/__init__.py:48: FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.layers", FutureWarning) /usr/local/miniconda3/envs/ocinet_npu/lib/python3.8/site-packages/timm/models/registry.py:4: FutureWarning: Importing from timm.models.registry is deprecated, please import via timm.models warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.models", FutureWarning) /root/OCINet-main/model/pvtv2.py:388: UserWarning: Overwriting pvt_v2_b0 in registry with model.pvtv2.pvt_v2_b0. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b0(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:398: UserWarning: Overwriting pvt_v2_b1 in registry with model.pvtv2.pvt_v2_b1. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b1(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:406: UserWarning: Overwriting pvt_v2_b2 in registry with model.pvtv2.pvt_v2_b2. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b2(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:414: UserWarning: Overwriting pvt_v2_b3 in registry with model.pvtv2.pvt_v2_b3. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b3(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:422: UserWarning: Overwriting pvt_v2_b4 in registry with model.pvtv2.pvt_v2_b4. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b4(PyramidVisionTransformerImpr): /root/OCINet-main/model/pvtv2.py:431: UserWarning: Overwriting pvt_v2_b5 in registry with model.pvtv2.pvt_v2_b5. This is because the name being registered conflicts with an existing name. Please check if this is not expected. class pvt_v2_b5(PyramidVisionTransformerImpr): Dropped keys (num_batches_tracked): 33 Missing keys: 0 Unexpected keys: 0 Missing sample: [] Unexpected sample: [] Saved: /root/OCINet-main/ocinet_352.onnx -rw-r--r-- 1 root root 115M Jan 5 11:34 ./ocinet_352.onnx

(ocinet_npu) root@davinci-mini:~/OCINet-main# atc \ --model=/root/OCINet-main/ocinet_352.onnx \ --framework=5 \ --output=ocinet_352 \ --input_format=NCHW \ --input_shape="input:1,3,352,352" \ --soc_version=Ascend310B4 \ --output_type=FP32 \ --precision_mode=allow_fp32_to_fp16 ATC start working now, please wait for a moment. /usr/local/Ascend/ascend-toolkit/latest/python/site-packages/tbe/dsl/unify_schedule/extract_image_patches_without_cbuf_schedule.py:317: SyntaxWarning: "is not" with a literal. Did you mean "!="? if _ is not 1: \