Skip to content

Commit 3b3c2ea

Browse files
Update imageio.imsave -> imageio.v3.imwrite
1 parent 74b60dd commit 3b3c2ea

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

README.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,20 +15,20 @@ Load a RAW file and save the postprocessed image using default [parameters](http
1515

1616
```python
1717
import rawpy
18-
import imageio
18+
import imageio.v3 as iio
1919

2020
path = 'image.nef'
2121
with rawpy.imread(path) as raw:
2222
rgb = raw.postprocess()
23-
imageio.imsave('default.tiff', rgb)
23+
iio.imwrite('default.tiff', rgb)
2424
```
2525

2626
Save as 16-bit linear image:
2727

2828
```python
2929
with rawpy.imread(path) as raw:
3030
rgb = raw.postprocess(gamma=(1,1), no_auto_bright=True, output_bps=16)
31-
imageio.imsave('linear.tiff', rgb)
31+
iio.imwrite('linear.tiff', rgb)
3232
```
3333

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

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

6565
## Installation

rawpy/_rawpy.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -877,7 +877,7 @@ cdef class RawPy:
877877
with open('thumb.jpg', 'wb') as f:
878878
f.write(thumb.data)
879879
elif thumb.format == rawpy.ThumbFormat.BITMAP:
880-
imageio.imsave('thumb.tiff', thumb.data)
880+
imageio.v3.imwrite('thumb.tiff', thumb.data)
881881
882882
:rtype: :class:`rawpy.Thumbnail`
883883
"""

rawpy/enhance.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -350,11 +350,11 @@ def save_dcraw_bad_pixels(path, bad_pixels):
350350
paths = [os.path.join(prefix, f) for f in testfiles]
351351
coords = find_bad_pixels(paths)
352352

353-
import imageio
353+
import imageio.v3 as iio
354354
raw = rawpy.imread(paths[0])
355355
if not os.path.exists('test_original.png'):
356356
rgb = raw.postprocess()
357-
imageio.imsave('test_original.png', rgb)
357+
iio.imwrite('test_original.png', rgb)
358358

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

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

379379
# TODO method 'mean' not implemented yet
380380

0 commit comments

Comments
 (0)