Borcherding commited on
Commit
0356a71
·
verified ·
1 Parent(s): 842e252

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +32 -1
README.md CHANGED
@@ -1,3 +1,34 @@
1
  ---
2
- license: apache-2.0
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ tags:
3
+ - depth-to-robot
4
+ - image-to-image
5
+ - cyclegan
6
  ---
7
+
8
+ # Depth2Robot GAN Model
9
+
10
+ This model transforms depth maps into robot-style images using CycleGAN.
11
+
12
+ ## Model Description
13
+
14
+ - This model was trained on depth maps and robot images.
15
+ - It converts grayscale depth maps to colorful robot-style imagery.
16
+ - Trained using CycleGAN architecture.
17
+
18
+ ## Usage
19
+
20
+ ```python
21
+ import torch
22
+ from huggingface_hub import hf_hub_download
23
+
24
+ # Download the model
25
+ model_path = hf_hub_download(repo_id="Borcherding/depth2AnythingCycleGAN_RobotsV2", filename="latest_net_G.pth")
26
+
27
+ # Load the model (you need to define the Generator class)
28
+ model = Generator()
29
+ model.load_state_dict(torch.load(model_path), strict=False)
30
+ model.eval()
31
+
32
+ # Use the model for inference
33
+ # ...
34
+ ```