19 lines
579 B
Python
19 lines
579 B
Python
from mflux.flux_tools.depth.depth_util import DepthUtil
|
|
from mflux.models.depth_pro.depth_pro import DepthPro
|
|
from mflux.ui.cli.parsers import CommandLineParser
|
|
|
|
|
|
def main():
|
|
# 0. Parse command line arguments
|
|
parser = CommandLineParser(description="Save a depth map of an input image")
|
|
parser.add_save_depth_arguments()
|
|
args = parser.parse_args()
|
|
|
|
# 1. Create and save the depth map
|
|
depth_pro = DepthPro(quantize=args.quantize)
|
|
DepthUtil.get_or_create_depth_map(depth_pro=depth_pro, image_path=args.image_path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|