nielsr HF staff commited on
Commit
29ba8c0
1 Parent(s): d964332

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -45,8 +45,9 @@ transform = Compose([
45
  PrepareForNet(),
46
  ])
47
 
48
- img = Image.open(...)
49
- image = transform({'image': np.array(image)})['image']
 
50
  image = torch.from_numpy(image).unsqueeze(0)
51
 
52
  depth = model(image)
 
45
  PrepareForNet(),
46
  ])
47
 
48
+ image = Image.open("...")
49
+ image = np.array(image) / 255.0
50
+ image = transform({'image': image})['image']
51
  image = torch.from_numpy(image).unsqueeze(0)
52
 
53
  depth = model(image)