Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,7 @@ This model transforms Gradient Magnitude maps into coral reef style images, and
|
|
39 |
```bash
|
40 |
# Clone the repository
|
41 |
git clone https://huggingface.co/Borcherding/CycleGAN_GradientMagnitude2CoralReef_Blend
|
42 |
-
cd
|
43 |
|
44 |
# Install dependencies
|
45 |
pip install torch torchvision gradio pyvirtualcam
|
@@ -164,7 +164,7 @@ def download_model(direction="depth2image"):
|
|
164 |
filename = "latest_net_G_B.pth"
|
165 |
|
166 |
model_path = hf_hub_download(
|
167 |
-
repo_id="Borcherding/
|
168 |
filename=filename
|
169 |
)
|
170 |
return model_path
|
@@ -243,7 +243,7 @@ def transform_image(input_image_path, direction="depth2image"):
|
|
243 |
## Model Checkpoints
|
244 |
|
245 |
The model checkpoints are available on Hugging Face:
|
246 |
-
- Repository: [Borcherding/Depth2RobotsV2_Annotations](https://huggingface.co/datasets/Borcherding/
|
247 |
- Files:
|
248 |
- `latest_net_G_A.pth` - Generator for Depth to Robot Image transformation
|
249 |
- `latest_net_G_B.pth` - Generator for Robot Image to Depth transformation
|
|
|
39 |
```bash
|
40 |
# Clone the repository
|
41 |
git clone https://huggingface.co/Borcherding/CycleGAN_GradientMagnitude2CoralReef_Blend
|
42 |
+
cd CycleGAN_GradientMagnitude2CoralReef_Blend
|
43 |
|
44 |
# Install dependencies
|
45 |
pip install torch torchvision gradio pyvirtualcam
|
|
|
164 |
filename = "latest_net_G_B.pth"
|
165 |
|
166 |
model_path = hf_hub_download(
|
167 |
+
repo_id="Borcherding/CycleGAN_GradientMagnitude2CoralReef_Blend",
|
168 |
filename=filename
|
169 |
)
|
170 |
return model_path
|
|
|
243 |
## Model Checkpoints
|
244 |
|
245 |
The model checkpoints are available on Hugging Face:
|
246 |
+
- Repository: [Borcherding/Depth2RobotsV2_Annotations](https://huggingface.co/datasets/Borcherding/GradientMagnitude2CoralReef_Annotations)
|
247 |
- Files:
|
248 |
- `latest_net_G_A.pth` - Generator for Depth to Robot Image transformation
|
249 |
- `latest_net_G_B.pth` - Generator for Robot Image to Depth transformation
|