interdigitalinc / hrfae Goto Github PK

How much time do you spend on 70,000 pictures while training?

@Xu-Yao Thanks you very much for the code you shared.Could you please provide the code train for other resolution pictures and your dataset about 50,000 face images you made.

Do anyone try to convert trained model to ONNX model, please?
Don't know how to use torch.export because there're multiple networks.
Hope can get help here, thanks.

I have verified that I have installed pytorch, when I type conda list I see the following:

pytorch 1.5.0 py3.7_cuda102_cudnn7_0 pytorch

however, when I try to run this code after following your instructions and setting up an environment I receive the following error:

Traceback (most recent call last):
File "test.py", line 11, in
import torch
ModuleNotFoundError: No module named 'torch'

Any help with this issue would be greatly appreciated.

Hi, will you provide a working docker image?

Hi, I would like to train the model, but I had the problem when convert the caffe model to pt file. Is that possible to provide the dex_imdb_wiki.caffemodel.pt file for me?

Thanks for your work very much.
there is a question that: when I test the pics prepared, i set '--target_age 5' , it can not generate faces of chilren well ?

While trying to create a virtual environment using the steps mentioned in README.md , I faced this issue:

ResolvePackageNotFound: - openssl==1.1.1=h7b6447c_0

Did anyone face a similar issue?

Hi bor:
when I try to run test.py, and when the code go into line 58:
trainer.load_checkpoint(log_dir + 'checkpoint')
then an error happend at here.
'''
raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
'''
this looks like means I must run these code on a cuda device.
or because of this test model is trained on a cuda device, I must build a new model which supported on CPU only?

Hi @Xu-Yao
Thanks for sharing this code.

I checked the paper, and found that you are solving data imbalance using StyleGAN generated images (Section 4.1).
But unfortunately, not in the paper or here, there is no information on how to access this dataset (age-equalized).

If it is possible, could you share this dataset, please.

Does anyone have the age tag for Celeba-HQ?

As I do not have access to Telecom resources anymore, the download link for the pre-trained model may soon be disabled. You can download the pre-trained checkpoint here: https://drive.google.com/file/d/11wK7cHOH--CWgNSbZucJ4nppon4BN9eJ/view?usp=share_link

could you please provide the dex_imdb_wiki.caffemodel.pt?

Hello:
I find some code that I think may be wrong
return nn.CrossEntropyLoss()(x, target_age)
https://github.com/InterDigitalInc/HRFAE/blob/master/trainer.py#L62

which should be
return nn.CrossEntropyLoss(x, target_age)
?

I don't know if it has some impact.

I have try run the test.py and download the pretrained model. but the result is bad, so I'm confused any steps is wrong?

Hello, may I ask you why the encode structure takes a long time, about 33s

We have:

# Input data
input_w: 1024
input_h: 1024

in configs/001.yml file. Can't we change it to other pixel formats, for example, 3780x6720 or retaining the original size of images? Please feel free to throw some light on why it can't accept the image as is.

I tried changing it to given example but the program was immediately killed.

It gives weird outputs when the size is changed to 1024x1024. Check the images. The first one is original, second one is the output by the model.

Hello,
looks like download weight URL died.
it returns HTTP ERROR 500 and get a 0 byte checkpoint file.
could you please fix it?
Thanks.

Hello there,

i have issue in smaller size ... changed width height in config to 256... and changed in image file also...

and getting output not proper ... can any one help me please ...

Originally posted by @kishanpanchotiya in #8 (comment)

Hi, I had the problem when convert the caffe model to pt file. Problem is" FileNotFoundError: [Errno 2] No such file or directory: 'protoc': 'protoc' "
Is that possible to provide the dex_imdb_wiki.caffemodel.pt file for me ?
Thanks.