neuralchen / SimSwap

An arbitrary face-swapping framework on images and videos with one single trained model!
Other
4.55k stars 895 forks source link

Module Not Found Error #436

Open joaoesp opened 1 year ago

joaoesp commented 1 year ago

When running python test_wholeimage_swapsingle.py I get this error:

Traceback (most recent call last): File "test_wholeimage_swapsingle.py", line 51, in model = create_model(opt) File "C:\Users\Joaoesp\Documents\SimSwap\SimSwap-main\models\models.py", line 18, in create_model model.initialize(opt) File "C:\Users\Joaoesp\Documents\SimSwap\SimSwap-main\models\fs_model.py", line 67, in initialize netArc_checkpoint = torch.load(netArc_checkpoint) File "C:\Users\Joaoesp\miniconda3\envs\simswap\lib\site-packages\torch\serialization.py", line 592, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "C:\Users\Joaoesp\miniconda3\envs\simswap\lib\site-packages\torch\serialization.py", line 851, in _load result = unpickler.load() ModuleNotFoundError: No module named 'models.arcface_models'

Anyone knows what might be causing the problem?

Here's a list of my packages:

Name Version Build Channel

blas 2.117 mkl conda-forge blas-devel 3.9.0 17_win64_mkl conda-forge brotli-python 1.0.9 py38hd3f51b4_9 conda-forge bzip2 1.0.8 h8ffe710_4 conda-forge ca-certificates 2023.7.22 h56e8100_0 conda-forge certifi 2023.7.22 pyhd8ed1ab_0 conda-forge charset-normalizer 3.2.0 pyhd8ed1ab_0 conda-forge colorama 0.4.6 pypi_0 pypi coloredlogs 15.0.1 pypi_0 pypi contourpy 1.1.0 pypi_0 pypi cuda-cccl 12.2.53 0 nvidia cuda-cudart 11.7.99 0 nvidia cuda-cudart-dev 11.7.99 0 nvidia cuda-cupti 11.7.101 0 nvidia cuda-libraries 11.7.1 0 nvidia cuda-libraries-dev 11.7.1 0 nvidia cuda-nvrtc 11.7.99 0 nvidia cuda-nvrtc-dev 11.7.99 0 nvidia cuda-nvtx 11.7.91 0 nvidia cuda-runtime 11.7.1 0 nvidia cudatoolkit 10.2.89 he40447d_12 conda-forge cycler 0.11.0 pypi_0 pypi decorator 4.4.2 pypi_0 pypi easydict 1.10 pypi_0 pypi filelock 3.12.2 pyhd8ed1ab_0 conda-forge flatbuffers 23.5.26 pypi_0 pypi fonttools 4.41.1 pypi_0 pypi freetype 2.12.1 h546665d_1 conda-forge humanfriendly 10.0 pypi_0 pypi idna 3.4 pyhd8ed1ab_0 conda-forge imageio 2.31.1 pypi_0 pypi imageio-ffmpeg 0.4.8 pypi_0 pypi importlib-resources 6.0.0 pypi_0 pypi insightface 0.2.1 pypi_0 pypi intel-openmp 2023.2.0 h57928b3_49496 conda-forge jinja2 3.1.2 pyhd8ed1ab_1 conda-forge joblib 1.3.1 pypi_0 pypi jpeg 8d h8ffe710_1 conda-forge kiwisolver 1.4.4 pypi_0 pypi lazy-loader 0.3 pypi_0 pypi lerc 4.0.0 h63175ca_0 conda-forge libblas 3.9.0 17_win64_mkl conda-forge libcblas 3.9.0 17_win64_mkl conda-forge libcublas 11.10.3.66 0 nvidia libcublas-dev 11.10.3.66 0 nvidia libcufft 10.7.2.124 0 nvidia libcufft-dev 10.7.2.124 0 nvidia libcurand 10.3.3.53 0 nvidia libcurand-dev 10.3.3.53 0 nvidia libcusolver 11.4.0.1 0 nvidia libcusolver-dev 11.4.0.1 0 nvidia libcusparse 11.7.4.91 0 nvidia libcusparse-dev 11.7.4.91 0 nvidia libdeflate 1.17 hcfcfb64_0 conda-forge libffi 3.4.2 h8ffe710_5 conda-forge libhwloc 2.9.2 nocuda_h15da153_1008 conda-forge libiconv 1.17 h8ffe710_0 conda-forge liblapack 3.9.0 17_win64_mkl conda-forge liblapacke 3.9.0 17_win64_mkl conda-forge libnpp 11.7.4.75 0 nvidia libnpp-dev 11.7.4.75 0 nvidia libnvjpeg 11.8.0.2 0 nvidia libnvjpeg-dev 11.8.0.2 0 nvidia libpng 1.6.39 h19919ed_0 conda-forge libsqlite 3.42.0 hcfcfb64_0 conda-forge libtiff 4.5.0 h8a3f274_0 libuv 1.44.2 hcfcfb64_1 conda-forge libwebp 1.3.1 hcfcfb64_0 conda-forge libwebp-base 1.3.1 hcfcfb64_0 conda-forge libxcb 1.13 hcd874cb_1004 conda-forge libxml2 2.11.4 hc3477c8_0 conda-forge libzlib 1.2.13 hcfcfb64_5 conda-forge m2w64-gcc-libgfortran 5.3.0 6 conda-forge m2w64-gcc-libs 5.3.0 7 conda-forge m2w64-gcc-libs-core 5.3.0 7 conda-forge m2w64-gmp 6.1.0 2 conda-forge m2w64-libwinpthread-git 5.0.0.4634.697f757 2 conda-forge markupsafe 2.1.3 py38h91455d4_0 conda-forge matplotlib 3.7.2 pypi_0 pypi mkl 2022.1.0 h6a75c08_874 conda-forge mkl-devel 2022.1.0 h57928b3_875 conda-forge mkl-include 2022.1.0 h6a75c08_874 conda-forge moviepy 1.0.3 pypi_0 pypi mpmath 1.3.0 pyhd8ed1ab_0 conda-forge msys2-conda-epoch 20160418 1 conda-forge networkx 3.1 pyhd8ed1ab_0 conda-forge ninja 1.11.1 h91493d7_0 conda-forge numpy 1.24.4 py38h1d91fd2_0 conda-forge onnx 1.14.0 pypi_0 pypi onnxruntime 1.15.1 pypi_0 pypi opencv-python 4.8.0.74 pypi_0 pypi openjpeg 2.5.0 ha2aaf27_2 conda-forge openssl 3.1.1 hcfcfb64_1 conda-forge packaging 23.1 pypi_0 pypi pillow 10.0.0 pypi_0 pypi pip 23.2.1 pyhd8ed1ab_0 conda-forge proglog 0.1.10 pypi_0 pypi protobuf 4.23.4 pypi_0 pypi pthread-stubs 0.4 hcd874cb_1001 conda-forge pthreads-win32 2.9.1 hfa6e2cd_3 conda-forge pyparsing 3.0.9 pypi_0 pypi pyreadline3 3.4.1 pypi_0 pypi pysocks 1.7.1 pyh0701188_6 conda-forge python 3.8.17 h4de0772_0_cpython conda-forge python-dateutil 2.8.2 pypi_0 pypi python_abi 3.8 3_cp38 conda-forge pytorch 1.8.0 py3.8_cuda10.2_cudnn7_0 pytorch pytorch-cuda 11.7 h16d0643_5 pytorch pytorch-mutex 1.0 cuda pytorch pywavelets 1.4.1 pypi_0 pypi requests 2.31.0 pyhd8ed1ab_0 conda-forge scikit-image 0.21.0 pypi_0 pypi scikit-learn 1.3.0 pypi_0 pypi scipy 1.10.1 pypi_0 pypi setuptools 68.0.0 pyhd8ed1ab_0 conda-forge six 1.16.0 pypi_0 pypi sympy 1.12 pyh04b8f61_3 conda-forge tbb 2021.10.0 h91493d7_0 conda-forge threadpoolctl 3.2.0 pypi_0 pypi tifffile 2023.7.10 pypi_0 pypi tk 8.6.12 h8ffe710_0 conda-forge torchaudio 0.8.0 py38 pytorch torchvision 0.9.0 py38_cu102 pytorch tqdm 4.65.0 pypi_0 pypi typing_extensions 4.7.1 pyha770c72_0 conda-forge ucrt 10.0.22621.0 h57928b3_0 conda-forge urllib3 2.0.4 pyhd8ed1ab_0 conda-forge vc 14.3 h64f974e_17 conda-forge vc14_runtime 14.36.32532 hfdfe4a8_17 conda-forge vs2015_runtime 14.36.32532 h05e6639_17 conda-forge wheel 0.41.0 pyhd8ed1ab_0 conda-forge win_inet_pton 1.1.0 pyhd8ed1ab_6 conda-forge xorg-libxau 1.0.11 hcd874cb_0 conda-forge xorg-libxdmcp 1.1.3 hcd874cb_0 conda-forge xz 5.2.6 h8d14728_0 conda-forge zipp 3.16.2 pypi_0 pypi zlib 1.2.13 hcfcfb64_5 conda-forge zstd 1.5.2 h12be248_7 conda-forge

98jhon98 commented 1 year ago

I have the same problem. Anyone can help?