Skip to content

Commit 6d2e90d

Browse files
committed
iuv shading return in pipeline
1 parent bda7a57 commit 6d2e90d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

intrinsic/pipeline.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -322,6 +322,7 @@ def run_pipeline(models, img_arr, stage=4, resize_conf=0.0, base_size=384, linea
322322
lr_shd = to2np(rough_shd.squeeze(0))
323323
wb_img = (lr_alb * get_brightness(lr_shd)).clip(0, 1)
324324

325+
results['iuv_shd'] = to2np(iuv_shd.squeeze(0))
325326
results['lr_clr'] = lr_clr
326327
results['lr_alb'] = lr_alb
327328
results['lr_shd'] = lr_shd

0 commit comments

Comments
 (0)