Skip to content

Commit c84b19f

Browse files
committed
fix overwrite issue for jpg images
1 parent 609dd84 commit c84b19f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ First, clone the github repo...
3737

3838
```bash
3939
git clone git@github.com:CompVis/depth-fm.git
40-
cd dfm
40+
cd depth-fm
4141
```
4242

4343
Then download the weights via

inference.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def main(args):
8282
depth = plt.get_cmap('magma')(depth, bytes=True)[..., :3]
8383

8484
# Save the depth map
85-
depth_fp = args.img.replace('.png', '-depth.png')
85+
depth_fp = args.img + '_depth.png'
8686
depth_img = Image.fromarray(depth)
8787
if depth_img.size != orig_res:
8888
depth_img = depth_img.resize(orig_res, Resampling.BILINEAR)

0 commit comments

Comments
 (0)