search

AshStuff / dissm

25 stars 7 forks source link

Missing the ”dataset.py” file in the "refinement" folder #1

Open yu-to-msk opened 2 years ago

yu-to-msk commented 2 years ago

Dear Ashwin Raju

Thank you for publically sharing the code of your wonderful research

I'm investigating your paper and code because I want to retry the DISSM segmentation experiments with my own dataset.

However, when I checked, I could not find the ”dataset.py” file in the "refinement" folder. ("data_augmentation.py" and "train.py" import funcions in "dataset.py") (I also checked "implicitshapes" folder but I couldn't find the SimpleRefineDataset, ConcatImPCA, and ComputeMask)

I'm sorry if I overlooked something.

Do you have plan to add a "dataset.py” file to the "refinement" folder of this repository to train refinment-model?

OR, is it possible for you to give me a quick rundown of the implementation of the "SimpleRefineDataset" class (in addition to ConcatImPCA and ComputeMask class)?

I know it's a big ask, and I appreciate you doing it for me.

Thank you.

jwutsetro commented 3 months ago

Hey, it seems that you where already able to train the PCA and rotation models. We are currently stuck at creating the correct config files for the rotation and pca models ( cfg.global_init_mtx in the augmentation pipeline). Would you maybe have some time to share how you proceeded with this?

Kindly, J