三行代码帮你自动画图

一周之前,百度开源的自动画图功能爆火,它可以三行就实现各种风格的自动画图。

首先安装包

    !pip install paddlehub

    然后就可以用三行代码画图了

      import paddlehub as hub 
      module = hub.Module(name="ernie_vilg") 
      results = module.generate_image(text_prompts=["波涛汹涌的大海"])

      图片

      这个包还提供了不同的画风,比如我们把ernie_vilg改成stable_diffusion

        module = hub.Module(name="stable_diffusion") results = module.generate_image(text_prompts=["波涛汹涌的大海"])

        事实上,paddlehub除了画图外还有很多功能,下面我们也顺带做点探索。

        1.  人像分割

        import paddlehub as hub
        module = hub.Module(name="humanseg_lite")
        res = module.segment(
            paths = ["human.png"],
            visualization=True,
            output_dir='humanseg_output')

        图片图片

        可以看到,效果还是不错的,如果图片是白底的话,效果应该更好。

        2. 人脸检测

          import paddlehub as hub
          module = hub.Module(name="ultra_light_fast_generic_face_detector_1mb_640")
          res = module.face_detection(
              paths = ["human.png"],
              visualization=True,
              output_dir='humanseg_output')

          图片

          3. 人体姿势检测

            import paddlehub as hub
            module = hub.Module(name="openpose_body_estimation")
            res = module.predict(
                img = "human.png",
                visualization=True,
                save_path='humanseg_output')

            图片

            此外,PaddleHub还具有一些Linux命令行工具。

            1.  hub list

            !hub list

            用于显示PaddleHub下安装的模型

            图片

            其中ernie_vilg、humanseg_lite等都是我们在《三行代码帮你自动画图》中分享过的模型。

            2. hub uninstall

            用于卸载模型,我们以humanseg_lite为例

              !hub uninstall humanseg_lite

              图片

              3. hub install

              用于安装模型,我们再把humanseg_lite装回去

                !hub install humanseg_lite

                图片

                4. hub show

                用于显示已安装模型的信息,还是以humanseg_lite为例

                  !hub show humanseg_lite

                  图片

                  5. hub run

                  用于运行模型,以lac模型为例

                    !hub run lac --input_text "今天是个好日子

                    图片

                    此外,还有hub version,hub config等。

                    © 版权声明
                    THE END
                    喜欢就支持一下吧
                    点赞0 分享