-
Notifications
You must be signed in to change notification settings - Fork 945
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
a small error in train.py #114
base: master
Are you sure you want to change the base?
Conversation
|
@destiny311 I have met the same problem, have you solved it? What is "--train_data_file“ |
@destiny311 @LucienXian I get the dataset by search "300W-LP" in Google, and you can find the download link of 300W-LP in the web http://www.cbsr.ia.ac.cn/users/xiangyuzhu/projects/3DDFA/main.htm. I generate groundtruth position map just using generate_posmap_300WLP |
Add loss weights and change image/label normalize method
Add loss weights and change image/label normalize method
https://github.com/jnulzl/PRNet/blob/bbdc3c95266aa4ce6379cd343121616244964872/train.py#L39
this should be 'label', instead of 'img', like as:
label_array = np.array(label,dtype=np.float32)
by the way, thanks for your train code!