cleargrasp icon indicating copy to clipboard operation
cleargrasp copied to clipboard

Which model is trained for Model checkpoints in README?

Open JinraeKim opened this issue 2 years ago • 0 comments

In READE.md:

Download the data: a) Model Checkpoints (0.9GB) - Trained checkpoints of our 3 deeplabv3+ models. b) Train dataset (Optional, 72GB) - Contains the synthetic images used for training the models. No real images were used for training. c) Val + Test datasets (Optional, 1.7GB) - Contains the real and synthetic images used for validation and testing.

And I realised that in the sample config, it is specified as drn:

  17 # Surface Normals Model
  18 normals:
  19   model: 'drn'  # Which model to use for inference. Possible values: ['unet', 'deeplab_resnet']
  20   pathWeightsFile: '../data/cleargrasp-checkpoints/surface_normals/checkpoint_normals.pth'  # Path to the checkpoint to be loaded
  21   inferenceHeight: 256
  22   inferenceWidth: 256

Is the Model checkpoints (provided in README) drn? Is drn different from unet or deeplab_resnet?

JinraeKim avatar Jul 20 '22 15:07 JinraeKim