Skip to content

Commit

Permalink
Update imageio.imsave -> imageio.v3.imwrite
Browse files Browse the repository at this point in the history
  • Loading branch information
mplough-kobold committed Aug 9, 2024
1 parent 74b60dd commit 3b3c2ea
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 10 deletions.
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,20 +15,20 @@ Load a RAW file and save the postprocessed image using default [parameters](http

```python
import rawpy
import imageio
import imageio.v3 as iio

path = 'image.nef'
with rawpy.imread(path) as raw:
rgb = raw.postprocess()
imageio.imsave('default.tiff', rgb)
iio.imwrite('default.tiff', rgb)
```

Save as 16-bit linear image:

```python
with rawpy.imread(path) as raw:
rgb = raw.postprocess(gamma=(1,1), no_auto_bright=True, output_bps=16)
imageio.imsave('linear.tiff', rgb)
iio.imwrite('linear.tiff', rgb)
```

Extract embedded thumbnail/preview image and save as JPEG:
Expand All @@ -44,7 +44,7 @@ if thumb.format == rawpy.ThumbFormat.JPEG:
f.write(thumb.data)
elif thumb.format == rawpy.ThumbFormat.BITMAP:
# thumb.data is an RGB numpy array, convert with imageio
imageio.imsave('thumb.jpeg', thumb.data)
iio.imwrite('thumb.jpeg', thumb.data)
```

Find bad pixels using multiple RAW files and repair them:
Expand All @@ -59,7 +59,7 @@ for path in paths:
with rawpy.imread(path) as raw:
rawpy.enhance.repair_bad_pixels(raw, bad_pixels, method='median')
rgb = raw.postprocess()
imageio.imsave(path + '.tiff', rgb)
iio.imwrite(path + '.tiff', rgb)
```

## Installation
Expand Down
2 changes: 1 addition & 1 deletion rawpy/_rawpy.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -877,7 +877,7 @@ cdef class RawPy:
with open('thumb.jpg', 'wb') as f:
f.write(thumb.data)
elif thumb.format == rawpy.ThumbFormat.BITMAP:
imageio.imsave('thumb.tiff', thumb.data)
imageio.v3.imwrite('thumb.tiff', thumb.data)
:rtype: :class:`rawpy.Thumbnail`
"""
Expand Down
8 changes: 4 additions & 4 deletions rawpy/enhance.py
Original file line number Diff line number Diff line change
Expand Up @@ -350,11 +350,11 @@ def save_dcraw_bad_pixels(path, bad_pixels):
paths = [os.path.join(prefix, f) for f in testfiles]
coords = find_bad_pixels(paths)

import imageio
import imageio.v3 as iio
raw = rawpy.imread(paths[0])
if not os.path.exists('test_original.png'):
rgb = raw.postprocess()
imageio.imsave('test_original.png', rgb)
iio.imwrite('test_original.png', rgb)

# A. use dcraw repair
# Note that this method fails when two bad pixels are direct neighbors.
Expand All @@ -365,7 +365,7 @@ def save_dcraw_bad_pixels(path, bad_pixels):
save_dcraw_bad_pixels(bad_pixels_path, coords)
rgb = raw.postprocess(bad_pixels_path=bad_pixels_path)
print('badpixel dcraw repair+postprocessing:', time.time()-t0, 's')
imageio.imsave('test_hotpixels_repaired_dcraw.png', rgb)
iio.imwrite('test_hotpixels_repaired_dcraw.png', rgb)

# B. use own repair function
# With method='median' we still consider each bad pixel separately
Expand All @@ -374,7 +374,7 @@ def save_dcraw_bad_pixels(path, bad_pixels):
repair_bad_pixels(raw, coords, method='median')
rgb = raw.postprocess()
print('badpixel repair+postprocessing:', time.time()-t0, 's')
imageio.imsave('test_hotpixels_repaired.png', rgb)
iio.imwrite('test_hotpixels_repaired.png', rgb)

# TODO method 'mean' not implemented yet

Expand Down

0 comments on commit 3b3c2ea

Please sign in to comment.