Skip to content

Commit

Permalink
readme file
Browse files Browse the repository at this point in the history
  • Loading branch information
rupalsaxena committed Jun 5, 2023
1 parent 58fb181 commit 0819c18
Showing 1 changed file with 15 additions and 5 deletions.
20 changes: 15 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ indoor scenes, accompanied by per-pixel ground truth labels. This data consists
Download the data from the link: [HypersimData](https://github.com/apple/ml-hypersim)

## RGB to Semantic Instances
**STEP 1**: Save rgb and groundtruth semantic instances in torch format using dataloader
**STEP 1**: Save rgb and groundtruth semantic instances in torch format using dataloader.
```
# navigate to hypersim dataloader
cd src/Monograph/dataloader/hypersim_pytorch
Expand All @@ -38,15 +38,25 @@ python3 save_hypersim_dataset.py
```
Data will be saved in your provided output path in torch format.

**STEP 2**: Train DeepLabv3 Resnet50 Model using transfer learning
**STEP 2**: Train DeepLabv3 Resnet50 Model using transfer learning.
```
# navigate to rgb-to-semantics directory
cd ../../rgb-to-semantics
```

**STEP 3**: Test and save the predicted semantic data
Update input path and trained model output path in config.py
Train the model
```
python3 transfer_learning.py
```
Once training is over, check the provided output path to see trained model.


**STEP 3**: Save predicted semantic data.
Update MODELPATH TESTDATAPATH, main_path in test_model.py file. Finally, run the file to get predicted semantic instances.
```
python3 test_model.py
```
Check the predicted semantic images in main_path provided in config.py file.

## RGB to Depth
Step 1: Load and save rgb and groundtruth depth in torch format using dataloader
Expand Down

0 comments on commit 0819c18

Please sign in to comment.