ndb796 / celeba-hq-face-identity-and-attributes-recognition-pytorch Goto Github PK
View Code? Open in Web Editor NEWCelebA HQ Face Identity and Attributes Recognition using PyTorch
CelebA HQ Face Identity and Attributes Recognition using PyTorch
Thank you for sharing the code for the gender and identity classifier. I wanted to know how to use the CelebA 64x64 dataset for training and testing.
Also I am facing the following issue when I run the script for testing on the CelebA 64x64 dataset:
/home/first_name.last_name/venv/lib/python3.8/site-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:100.)
return torch._C._cuda_getDeviceCount() > 0
Test dataset size: 32426
Class names: ['images']
Traceback (most recent call last):
File "ada_celeba_gender-classifier-modified.py", line 84, in <module>
model.load_state_dict(torch.load(save_path))
File "/home/first_name.last_name/venv/lib/python3.8/site-packages/torch/serialization.py", line 595, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/first_name.last_name/venv/lib/python3.8/site-packages/torch/serialization.py", line 764, in _legacy_load
magic_number = pickle_module.load(f, **pickle_load_args)
_pickle.UnpicklingError: invalid load key, '#'.
I request you to guide me.
Regrads
Prabhav
Hey,
I really like your implementation of the identity recognition. However, you only grouped the identities under numbers.
Do you have a list where you also assign names to the numbers?
Otherwise very great work.
A declarative, efficient, and flexible JavaScript library for building user interfaces.
๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. ๐๐๐
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google โค๏ธ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.