CRM Model Training

This repository contains the necessary files and instructions for setting up the environment and training the CRM model. This repository is specifically designed to train 3D geometry ie CRM model, as I am deliberately inputting RGB images (manually captured) and CCM images (previously generated).

Setup

1. Clone the CRM repository: git clone https://github.com/thu-ml/CRM
2. Follow the steps provided in the repository to set up the environment for this project.

Adding the Model

1. Add model2.py to the directory: model/crm
2. Update the __init__.py file in the model directory with the following code: from model.crm.model2 import CRM as train_crm

3. Add the rest of the necessary files to the directory as instructed.

   [update 08-04-2024]

   • made changes as suggested https://github.com/thu-ml/CRM/issues/3#issuecomment-2039220300
   • added sample data

   [update 10-04-2024]

   • added missing files and code as in issue https://github.com/SanketDhuri/crm_3d_training/issues/2#issue-2232861468

Contribution Feel free to contribute to this project by submitting pull requests or opening issues for any bugs or enhancements.

Acknowledgement

• CRM
• nvdiffrast
• GET3D