sha2nkt / deco

Estimate vertex-level 3D human-scene and human-object contacts across the full body mesh
https://deco.is.tue.mpg.de/
Other
63 stars 6 forks source link

Training with DAMON dataset, and using RICH dataset images #16

Closed EnnaSachdeva closed 7 months ago

EnnaSachdeva commented 7 months ago

Hi, Thank you for this valuable work and your contributions.

I get this error when training the model with the given configs/cfg_train.yml file with all 3 datasets: rich, damon and prox. Could you please suggest the resolution?

abs

If I train with damon only, I get this error.

damon

ac5113 commented 7 months ago

Hello,

For training using RICH and PROX, you would have to download the data from their corresponding websites (RICH, PROX). The error likely occurs since the data is missing, and we do not provide the npzs or the data we have used for training.

As for DAMON, we have updated the download section to provide the links for the 2D polygon contacts from HOT. Apologies for this oversight. With these files downloaded, the PAL loss term should work again.

ac5113 commented 7 months ago

I shall close this issue if you're able to train the model using the dataset now.

EnnaSachdeva commented 6 months ago

@ac5113 , with the following changes in config_train.py which means it uses only damon data for training, it still gives the errors related to RICH dataset.

image ` image

ac5113 commented 6 months ago

You should remove 'rich' and 'prox' altogether from the DATASETS list. The way the loader was written, all the mentioned datasets are loaded and then the dataloader is created by picking from the datasets according to the mentioned probabilities.

DATASETS: ['damon'] would work

EnnaSachdeva commented 6 months ago

It works now. Thank you so much!

Sharpiless commented 6 months ago

Hello, awesome work!

May I ask you how to prepare RICH and PROX for training?

Sharpiless commented 6 months ago

One more question, do the results in the paper trained on seperate or mixed datasets?

ac5113 commented 6 months ago

Hello, awesome work!

May I ask you how to prepare RICH and PROX for training?

You can use the instructions specified here to prepare the RICH and PROX datasets for training.

One more question, do the results in the paper trained on seperate or mixed datasets?

The results are on a mixture of RICH, PROX and DAMON datasets

Sharpiless commented 6 months ago

Thanks!

Sharpiless commented 6 months ago

Sorry for bothering you again for "DAMON challenge: 3D contact prediction from 2D images". Should I provide a result with higher Geo_Err as the arrow indicates? In https://codalab.lisn.upsaclay.fr/competitions/17561#results, higher Geo_Err leads to a higher ranking. But I think this might be a typo.

ac5113 commented 6 months ago

The arrow seems to only be for sorting purposes. Results with a lower geodesic error are better since the metric measures the deviation between the predicted and gt contact vertices

The results are compared according to the criteria explained on the website as elucidated by @sha2nkt below

Sharpiless commented 6 months ago

Thanks again. Any evaluation results of DECO trained on DAMON dataset only? Considering the large storage space occupied by the other two datasets, we plan to train with only a single dataset for the competition.

ac5113 commented 6 months ago

There was no training done on DAMON only, unfortunately. Thus we do not have these results. However, you are free to use any other additional training data if RICH and PROX are too bulky.

sha2nkt commented 6 months ago

@Sharpiless The ranking order on the CodaLab server is not meant to be the final ranking. As indicated on the challenge website, after the competition ends, we will average the rankings on "F1 Score" and "Geo Error" to determine the winners.