Comments (5)
You need to add the weights files from your screenshot somewhere and fix the paths:
- "epoch59.pth" https://drive.google.com/file/d/1Jb97z5Z5ca61-6W2RDOK0e2w_RlbeWgS/view?usp=drive_link
- "FaceBoxesV2.pth" https://github.com/jhb86253817/PIPNet/blob/master/FaceBoxesV2/weights/FaceBoxesV2.pth
from celebbasis.
We use PIPNet to align and crop the face, but missed their weight files.
You may visit PIPNet repo to download the missing weight files or using the files provided by @justindujardin .
These files can also be downloaded from our Baidu Yun Drive with extracting code ygss
.
We'll fix the hardcoded file paths
ASAP!
from celebbasis.
Just updating the paths to relative paths (after putting the weight files there) works:
evaluation/face_align/PIPNet/lib/tools.py line 60:
weight_file = 'evaluation/face_align/PIPNet/weights/epoch59.pth'
state_dict = torch.load(weight_file, map_location=device)
net.load_state_dict(state_dict)
detector = FaceBoxesDetector(
"FaceBoxes",
"evaluation/face_align/PIPNet/weights/FaceBoxesV2.pth",
use_gpu=True,
device="cuda:0",
)
from celebbasis.
This entire section of code is also full of hardcoded paths:
https://github.com/ygtxr1997/CelebBasis/blob/main/scripts/stable_txt2img.py#L272
from celebbasis.
Just updating the paths to relative paths (after putting the weight files there) works:
evaluation/face_align/PIPNet/lib/tools.py line 60:
weight_file = 'evaluation/face_align/PIPNet/weights/epoch59.pth' state_dict = torch.load(weight_file, map_location=device) net.load_state_dict(state_dict) detector = FaceBoxesDetector( "FaceBoxes", "evaluation/face_align/PIPNet/weights/FaceBoxesV2.pth", use_gpu=True, device="cuda:0", )
- Code updated, 82fdc0e.
This entire section of code is also full of hardcoded paths: https://github.com/ygtxr1997/CelebBasis/blob/main/scripts/stable_txt2img.py#L272
- For
stable_txt2img.py#L272
, these hardcoded paths are not used under the default setting. We consider removing these codes or adding the relevant testing options.
from celebbasis.
Related Issues (19)
- Code release timeline? HOT 2
- Colab link? HOT 3
- very slow start_align.py unless change is made to DataLoader instance HOT 2
- no .pickle file for /infer_images folder? HOT 3
- What are the CPU RAM requirements? VRAM requirements? HOT 1
- Missing /taming folder in the root directory (submodule needed?) HOT 1
- device mismatch (indices are on gpu) HOT 2
- Does this compare to roop?
- RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory HOT 1
- raise MisconfigurationException(f"No {loader_name}() method defined to run Trainer.{trainer_method}.") HOT 3
- Replacing sd checkpoint with newer version HOT 2
- What is the VRAM usage of CeleBasis?
- Amazing work and easy to convert the result for webui use!!! HOT 4
- confusing about the shape of B? HOT 4
- sd-v1-4-full-ema.ckpt HOT 1
- Celebs names
- finetuning with multiple persons
- The source images and prompt for evaluation.
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 celebbasis.