Skip to main content

A small example package of demosaic-raw and cvisp, for our team

Project description

""" This is a demo for cvisp, 开发板、相机等拍摄设备取到的RAW with metadata,可以直接用该工具包进行demosaic,得到RGB图像,以便于展示。 - pip install cvisp - from cvisp.utils import Demosaic_RAW

"""

""" 三种模式,分别为:预览raw图,推理GDIP,预览demosaic(自用) run_demosaic(): return the demosaiced image, (HWC, u16, bmp) run_isp_with_meta(): return the rgb image , (HWC, u8, bmp) run_model(weight_path = './your.pt', model = GDIP(), device = 'cuda:0'): return the output of your model, (HWC, u8, bmp)

完整传参说明:
- Demosaic_RAW(mode : str = None, input_data_path : str = None, show_meta_data : bool = True, 
show_image : bool = False, save_image : bool = True, result_output_dir : str = None):
    mode: defatult = 'simple_dm', choices = ['simple_dm', 'opencv_dm']
    input_data_path: raw图片路径,支持单张图片,如:'./test.dng'
    show_meta_data: show the meta data of the input data, default = True,你可以选择是否展示meta data
    show_image: show the output image, default = False,你可以选择是否展示输出图像
    save_image: save the output image, default = True,你可以选择是否保存输出图像,保存路径为result_output_dir
    result_output_dir: the output dir of the result image, default = './demosaic/',你可以选择保存路径

"""

if name == 'main': for im in glob.glob(image_path_list + '*.dng'): demosaicd_raw = Demosaic_RAW(input_data_path = im).run_model(weight_path = './your.pt', model = GDIP(), device = 'cuda:0') # rgb = Demosaic_RAW(input_data_path = im).run_isp_with_meta() # gdip_out = Demosaic_RAW(input_data_path = im).run_model(weight_path = './your.pt', model = GDIP(), device = 'cuda:0')

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

cvisp-0.1.3.tar.gz (6.9 kB view hashes)

Uploaded Source

Built Distribution

cvisp-0.1.3-py3-none-any.whl (6.9 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page