Borcherding commited on
Commit
de65fbb
·
verified ·
1 Parent(s): 8dd4b84

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 cycleGAN_Depth2RobotsV2
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/CycleGAN_Depth2RobotsV2_Blend",
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/Depth2RobotsV2_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
 
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