對于赫本相信大家都非常熟悉了，絕對是一代女神，今天我們就來爬取女神的近千張美照，在一飽眼福的同時，還可以學習下如何做網站，對于老舊的黑白照片，還有一鍵上色功能可以玩，真是一舉多得

照片爬取

我們選擇的目的網站是時光網，話不多說，直接開整

http://people.mtime.com/951204/images

網站基本是沒有反爬蟲設定的，而且網頁結構也比較簡單，我們直接上代碼

先撰寫一個搜集所有圖片 URL 的函式以及一個保存函式

def download_url():
    images = []
    res = requests.get("http://front-gateway.mtime.com/library/person/imageAll.api?personId=951204")
    data = res.json().get("data")
    if data:
        images_list = data.get("images")
        for image in images_list:
            images.append(image.get("imageOrigin"))
    return images

def save_url(url):
    with open("url.csv", "w") as f:
        for i in url:
            f.write(i + "\n")

然后就可以決議得到的檔案，逐個爬取圖片了

def download_pic():
    with open("url.csv", "r") as f:
        url_list = f.readlines()
    print(url_list)
    for url in url_list:
        res = requests.get(url.replace("\n", ""))
        file = open("pic/" + url.replace("\n", "").split("/")[-1], "wb")
        file.write(res.content)
        file.close()

這要我們就完成了女神照片的采集作業

我們可以看到，有相當一部分照片都是黑白照，這大大的影響了我們的審美體驗，下面就進行上色處理

黑白照上色

我們這里選擇使用百度云提供的服務，大概有一年三百多次的免費使用次數

要使用百度云的功能，需要先登錄，然后再創建專案，具體這里就不展開說了，如果有小伙伴在這里遇到問題，可以加我微信私聊

我們創建好應用之后，可以看到有兩個 key 值，這是我們在呼叫百度介面是要用到的，分別是 API Key 和 Secret Key

下面我們就來測驗一下上色的功能，我們先選取一張黑白照片

嗯，要舔屏了（🐕）

下面開始編碼，先獲取 access token

import requests

# client_id 為官網獲取的AK， client_secret 為官網獲取的SK
host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=Yourapikey&client_secret=Yoursecretkey'
response = requests.get(host)
if response:
    print(response.json())
access_token = response.json()['access_token']

接下來我們構造上色介面的請求訊息

import base64

# 黑白影像上色
request_url = "https://aip.baidubce.com/rest/2.0/image-process/v1/colourize"
# 二進制方式打開圖片檔案
f = open('he1.jpg', 'rb')
img = base64.b64encode(f.read())
params = {"image": img}
request_url = request_url + "?access_token=" + access_token
headers = {'content-type': 'application/x-www-form-urlencoded'}
response = requests.post(request_url, data=params, headers=headers)

這要如果不出意外的話，我們就可以保存回傳的彩色圖片了

# base64編碼轉圖片
img = base64.b64decode(response.json()['image'])
print(img)
file = open('result.jpg', 'wb')
file.write(img)
file.close()

我們來看下效果

還是相當不錯的嘛，感覺女神的顏值也瞬間增加了幾分

黑白照片上色的流程跑通了，下面我們就開始制作網站，通過頁面的按鈕來進行圖片選擇已經上色請求

上色網站制作

還是使用 flask 搭建 web 服務

from flask import Flask, render_template, jsonify, request
import requests
import base64
import time


app = Flask(__name__)


@app.route("/")
def index():
    return render_template('index.html')


if __name__ == '__main__':
    app.run(debug=True)

接下來我們看下 index 當中的頁面布局，整體效果如下

左側用于選擇黑白圖片，然后點擊”圖片上色“按鈕后，在右側會顯示上色之后的圖片

下面來看看看部分 index 檔案代碼，主體使用 bootstrap 搭建

<div class="row">
        <div class="col-xs-6 col-md-6">
            <div class="thumbnail">
                <img id="pic" src="/static/tmp_pic/he1.jpg" alt="...">
                <div class="caption">
                    <form id="imageForm" enctype="multipart/form-data" method="post" οnsubmit="return false">
                        <a href="javascript:;" class="file">選擇檔案
                            <input type="file" name="" id="intro_pic">
                        </a>
                    </form>
                    <a href="javascript:;" class="file">圖片上色
                        <input type="submit" name="" id="add_colorize">
                    </a>
                </div>
            </div>
        </div>
        <div class="col-xs-6 col-md-6">
            <div class="thumbnail">
                <img id="colorize_pic" src="/static/tmp_pic/coming.png" alt="...">
                <div class="caption">
                </div>
            </div>
        </div>
    </div>

在上面的 input 組件中，呼叫了函式 intro_pic，我們來看下代碼

// 顯示圖片
    $("#intro_pic").on("change", function () {
        var filePath = $(this).val();//讀取圖片路徑

        var fr = new FileReader();//創建new FileReader()物件
        var imgObj = this.files[0];//獲取圖片
        fr.readAsDataURL(imgObj);//將圖片讀取為DataURL

        if (filePath.indexOf("jpg") !== -1 || filePath.indexOf("JPG") !== -1 || filePath.indexOf("PNG") !== -1 || filePath.indexOf("png") !== -1) {
            fr.onload = function () {
                $("#pic").attr('src', this.result);
            };
        }
    });

這部分代碼是用來處理圖片顯示的，既當我們上傳好圖片之后，就回顯到頁面上

再來看下 add_colorize 函式

$("#add_colorize").click(function () {
        var img = document.getElementById("pic");
        var path = img.src;
        var filename;
        if (path.indexOf("/") > 0)//如果包含有"/"號 從最后一個"/"號+1的位置開始截取字串
        {
            filename = path.substring(path.lastIndexOf("/") + 1, path.length);
        }
        else {
            filename = path;
        }
        var pic_src = $("#pic").attr('src');
        $.post("http://127.0.0.1:5000/download_pic/" + filename, {data: pic_src}, function (data, status) {
            fire_baidu(data);
        });
    });

這里把前端的 base64 型別的圖片資料傳遞給后端，然后通過后端的處理之后，再直接請求百度 API 介面

前端代碼就這樣，我們再來看看后端的代碼

@app.route("/save_colorize_pic", methods=["POST"])
def save_colorize_pic():
    data = request.values.get("image")
    try:
        img = base64.b64decode(data)
        img_path = "static/download_old_pic/" + str(int(time.time())) + 'result.jpg'
        file = open(img_path, 'wb')
        file.write(img)
        file.close()
        return jsonify(img_path)
    except Exception as e:
        print(e)
        return jsonify("error"), 400

上面的代碼從前端拿到上色之后的圖片資料，然后保存到本地

最后我們看一個最終的效果圖吧

看起來還不錯哦，快來一起實踐下吧

下面留言獲取完整代碼哦