Skip to content

Commit

Permalink
Update weight file download link to s3 url
Browse files Browse the repository at this point in the history
  • Loading branch information
yuguang-li committed Jul 11, 2023
1 parent 324fcf7 commit f6c34ab
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ For example, `SAVE_DIR` could be `/mnt/data/johnlam/zind_bridgeapi_2021_10_05`,
Make sure you are within the conda environment (`salve-v1`).

**Download ModifiedHorizonNet (MHNet) predictions.**
Download and unzip the custom `ModifiedHorizonNet` predictions from [Google Drive here](https://drive.google.com/file/d/16JuxBgg368RL7dSpPjF9kKocLbo2ekAt/view?usp=sharing).
Download and unzip the custom `ModifiedHorizonNet` predictions from [Google Drive here](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/data/ZInD_HorizonNet_predictions.tar.gz).



Expand Down Expand Up @@ -187,13 +187,13 @@ We release 7 pretrained models:

| W/D/O + Layout Source | Input Modalities | #Tours used for training | Arch. | Model filename (md5sum) |
| :-------------------: | :------------------: | :----------------------: | :---: | :--------------------: |
| MHNet | Ceiling + Floor RGB | 435 | ResNet-152 | [1200ffbe47d836557d88fef052952337.pth](https://drive.google.com/file/d/1MHlG2dJ9Vr0VwxlV7OH0ZabsZP8YQ2s0/view?usp=sharing) |
| MHNet | Ceiling + Floor RGB | 587 | ResNet-152 | [9fcbb628bd5efffbdcc4ce55a9eb380d.pth](https://drive.google.com/file/d/1BxPiwUCPuV8pgnVwpUzlWTkA_mCgMfx5/view?usp=sharing) |
| MHNet | Ceiling RGB only | 587 | ResNet-152 | [5c64123c134b829dd99beb3684582f61.pth](https://drive.google.com/file/d/1Qx1PwHFRHn62AtJg7uFFbBvv4gP2WjVM/view?usp=sharing) |
| MHNet | Floor RGB only | 587 | ResNet-152 | [a063532031f83aec97289466943bf52d.pth](https://drive.google.com/file/d/1WBidn9z6InHUgQbJ0MsY9OAZhcyCTSaL/view?usp=sharing) |
| MHNet | Rasterized Layout (Floor)| 877 | ResNet-152 | [6ac3f3e5fe6fa3d4bfae7c124d7787b3.pth](https://drive.google.com/file/d/1HwUO-ILcnUXfXhq1rlTGcmrbOMB9Owna/view?usp=sharing) |
| GT W/D/O + GT Layout | Ceiling + Floor RGB | 350 | ResNet-152 | [301f920ec795b9966aebc2367544d234.pth](https://drive.google.com/file/d/1BdrdYAXYFHn4EGxNYCish68GaDo3fvx3/view?usp=sharing) |
| GT W/D/O + GT Layout | Ceiling + Floor RGB | 817 | ResNet-152 | [b1198bad27aecb8a19f884abc920a731.pth](https://drive.google.com/file/d/1OuRI2wUi9wXZNux-mIYyxuqxp7J6Pn20/view?usp=sharing) |
| MHNet | Ceiling + Floor RGB | 435 | ResNet-152 | [1200ffbe47d836557d88fef052952337.pth](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/models/1200ffbe47d836557d88fef052952337.pth) |
| MHNet | Ceiling + Floor RGB | 587 | ResNet-152 | [9fcbb628bd5efffbdcc4ce55a9eb380d.pth](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/models/9fcbb628bd5efffbdcc4ce55a9eb380d.pth) |
| MHNet | Ceiling RGB only | 587 | ResNet-152 | [5c64123c134b829dd99beb3684582f61.pth](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/models/5c64123c134b829dd99beb3684582f61.pth) |
| MHNet | Floor RGB only | 587 | ResNet-152 | [a063532031f83aec97289466943bf52d.pth](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/models/a063532031f83aec97289466943bf52d.pth) |
| MHNet | Rasterized Layout (Floor)| 877 | ResNet-152 | [6ac3f3e5fe6fa3d4bfae7c124d7787b3.pth](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/models/6ac3f3e5fe6fa3d4bfae7c124d7787b3.pth) |
| GT W/D/O + GT Layout | Ceiling + Floor RGB | 350 | ResNet-152 | [301f920ec795b9966aebc2367544d234.pth](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/models/301f920ec795b9966aebc2367544d234.pth) |
| GT W/D/O + GT Layout | Ceiling + Floor RGB | 817 | ResNet-152 | [b1198bad27aecb8a19f884abc920a731.pth](https://files-zillowstatic-com.s3.us-west-2.amazonaws.com/research/public/StaticFiles/salve/models/b1198bad27aecb8a19f884abc920a731.pth) |


## Training a model
Expand Down

0 comments on commit f6c34ab

Please sign in to comment.