Comments (2)
Hi, @nagibator-hero!
The first thing you need to do is to determine the version of the CUDA with the command nvcc --version
.
Then install the correct version of mxnet with source 1 or source 2.
requirements.txt corresponds to CUDA 10.1.
To find the correct version of mxnet you can look at #56, #60 or try solution with docker #51.
from ghost.
For Windows
-
install anaconda, visual studio tools 2019, 2022 with c++ development checked
Cuda v10.1 here
Cudnn v8.0.5 here
Extract cudnn, copy all and paste to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
-
open up anaconda prompt, conda create -n sber python=3.6
-
conda activate sber
-
cd F:\AI\ghost (where you want to download)
-
git clone https://github.com/sberbank-ai/sber-swap.git
cd sber-swap
git submodule init
git submodule update(one by one)
-
pip install -r requirements.txt
if error, change requests==2.18.4
-
download models and move to the right directory
here -
Open model_zoo.py from C:\Users(username)\anaconda3\envs\sber\Lib\site-packages\insightface\model_zoo and edit line 23
def get_model(self):
session = onnxruntime.InferenceSession(self.onnx_file, providers=none)to
def get_model(self):
session = onnxruntime.InferenceSession(self.onnx_file, providers=['TensorrtExecutionProvider', 'CUDAExecutionProvider']) -
test and enjoy
python inference.py --source_paths {PATH_TO_IMAGE} --target_faces_paths {PATH_TO_IMAGE} --target_video {PATH_TO_VIDEO}
from ghost.
Related Issues (20)
- Help
- Серые лица HOT 2
- Colab no longer working HOT 5
- No module named 'insightface' HOT 1
- errors occure when the input video contains some no-face fragments
- google cloud is closed HOT 1
- Google Cloud is unavailable and banned! ! HOT 3
- Where
- Juanes
- TypeError: RetinaFace.detect() got an unexpected keyword argument 'threshold'
- Output is black blob on face of target image -- tips for debug?
- Suspicious/malicious file with binary incoded as base64 ofuscating as png image HOT 1
- Error when loading "Preparation" section HOT 1
- colab error OSError: libcudart.so.11.0: cannot open shared object file: No such file or directory HOT 1
- what is a model license? HOT 3
- Inconsistent requirements HOT 1
- Vgg2 dataset cannot be downloaded HOT 2
- How to Only Save the Best Model During Training
- Requirements.txt
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from ghost.