Video_Scene_Search_Image_Text/video_rest.py at master · rand0wn/Video_Scene_Search_Image_Text · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
"""REST API for Video Training and Search"""

from flask import Flask, render_template
from flask import jsonify
import video_process

app = Flask(__name__)


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


# Train Video
@app.route("/train/<path:path>")
def train_video(path):
    path = "/" + str(path)[0:len(str(path)) - 1]
    # Train Videos
    video_obj = video_process.Video(path, 10, 0)
    return video_obj.train_videos()


# Search Similar Scene(Image)
@app.route("/isearch/<path:path>")
def search_image_scene(path):
    # Search From Image
    path = "/"+ str(path)[0:len(str(path))-1]
    return jsonify(video_process._ext_img_idx(path))


# Search Similar Scene(Text)
@app.route("/tsearch/<string:text>")
def search_text_scene(text):
    # Search From Text
    return jsonify(video_process._text_idx(text))

if __name__ == "__main__":
    app.run(host='localhost', port=5003)