The celeba-hq-face-identity-and-attributes-recognition-pytorch from ndb796

celeba-hq-face-identity-and-attributes-recognition-pytorch's People

Contributors

Stargazers

Watchers

celeba-hq-face-identity-and-attributes-recognition-pytorch's Issues

Query: CelebA 64x64 Dataset

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

Facial_Identity_Classification Names

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.

Recommend Projects