はじめに
前回のcameraサンプルを改良して動画を再生できるようにしたいと思います。
また、画面に日本語を表示しようとすると文字化けするので表示出来るようにします。
ファイルコピー
camera.pyをコピーしてvideo.pyを作成します。
日本語表示
日本語対応のフォントに変更することで表示可能となります。以下はWindowsにインストールされている「MS ゴシック」の設定です。
|
1 2 |
from kivy.core.text import LabelBase, DEFAULT_FONT LabelBase.register(DEFAULT_FONT, "c:/windows/fonts/msgothic.ttc") |
画面改良
・ファイルダイアログを利用して動画ファイルを選択してファイル名を表示するように修正
・Face、Mosaicを車、ナンバーに修正
その他、Playを再生 Captureを保存にしました。
改良したソースファイル
保存先は、D:\PyDev\GuiKivyです。
仮想環境のアクティブ化を忘れずに実行します。env\Scripts\activate
|
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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 |
from kivy.config import Config Config.set('graphics', 'width', '800') Config.set('graphics', 'height', '640') from kivy.app import App from kivy.lang import Builder from kivy.uix.boxlayout import BoxLayout import time import cv2 from kivy.uix.image import Image from kivy.graphics.texture import Texture from kivy.properties import StringProperty, ObjectProperty from kivy.clock import Clock import os from kivy.uix.floatlayout import FloatLayout from kivy.uix.popup import Popup from kivy.uix.textinput import TextInput # 日本語表示対応 from kivy.core.text import LabelBase, DEFAULT_FONT LabelBase.register(DEFAULT_FONT, "c:/windows/fonts/msgothic.ttc") Builder.load_string(''' <CameraClick>: orientation: 'vertical' padding: 0 spacing: 1 size_hint_y: None BoxLayout: orientation: 'horizontal' padding: 0 spacing: 1 Image: id: camera size_hint: None, None size: 640, 480 BoxLayout: orientation: 'vertical' ToggleButton: text: '車' on_press: root.face() size_hint_y: None height: '48dp' ToggleButton: text: 'ナンバー' on_press: root.mosaic() size_hint_y: None height: '48dp' BoxLayout: orientation: 'horizontal' padding: 0 spacing: 1 size_hint_y: None height: '48dp' TextInput: id: txtFName text: '' multiline: False Button: text: 'ファイル選択' on_press: root.fileSelect() ToggleButton: text: '再生' on_press: root.play() size_hint_y: None height: '48dp' Button: text: '保存' size_hint_y: None height: '48dp' on_press: root.capture() <LoadDialog>: BoxLayout: size: root.size pos: root.pos orientation: 'vertical' FileChooserListView: id: filechooser BoxLayout: size_hint_y : None height : 30 Button: text: 'キャンセル' on_release: root.cancel() Button: text: '読み込み' on_release: root.load(filechooser.path, filechooser.selection) ''') class LoadDialog(FloatLayout): load = ObjectProperty(None) cancel = ObjectProperty(None) class CameraClick(BoxLayout): image_texture = ObjectProperty(None) image_capture = ObjectProperty(None) def fileSelect(self): content = LoadDialog(load = self.load, cancel = self.dismiss_popup) self._popup = Popup( title="動画ファイル選択中", content=content, size_hint=(0.9,0.9)) self._popup.open() def load (self, path, filename): txtFName = self.ids['txtFName'] txtFName.text = filename[0] self.dismiss_popup() def dismiss_popup(self): self._popup.dismiss() def face(self): global flgFace flgFace = not flgFace def mosaic(self): global flgMosaic flgMosaic = not flgMosaic def play(self): global flgPlay flgPlay = not flgPlay if flgPlay == True: txtFName = self.ids['txtFName'] self.image_capture = cv2.VideoCapture(txtFName.text) Clock.schedule_interval(self.update, 1.0 / 30) else: Clock.unschedule(self.update) self.image_capture.release() def update(self, dt): ret, frame = self.image_capture.read() if ret: if flgFace == True or flgMosaic == True: # カスケードファイルを指定して検出器を作成 cascade_file = "haarcascade_frontalface_alt.xml" cascade = cv2.CascadeClassifier(cascade_file) # カメラ映像をグレースケールに変換 frame_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # 顔検出 face_list = cascade.detectMultiScale(frame_gray, minSize=(30, 30)) if len(face_list) > 0: red = (0, 0, 255) for (x, y, w, h) in face_list: if flgFace == True: frame = cv2.rectangle(frame, (x, y), (x+w, y+h), red, thickness = 2) if flgMosaic == True: frame = self.procMosaic(frame, (x, y, x+w, y+h), 5) # カメラ映像を上下反転 buf = cv2.flip(frame, 0) # カメラ映像を上下左右反転 #buf = cv2.flip(frame, -1) image_texture = Texture.create(size=(frame.shape[1], frame.shape[0]), colorfmt='bgr') image_texture.blit_buffer(buf.tostring(), colorfmt='bgr', bufferfmt='ubyte') camera = self.ids['camera'] camera.texture = image_texture def capture(self): ''' Function to capture the images and give them the names according to their captured time and date. ''' camera = self.ids['camera'] timestr = time.strftime("%Y%m%d_%H%M%S") camera.export_to_png("IMG_{}.png".format(timestr)) print("Captured") def procMosaic(self, img, rect, size): # モザイクをかける領域を取得 (x1, y1, x2, y2) = rect w = x2 - x1 h = y2 -y1 i_rect =img[y1:y2, x1:x2] # 一度縮小して拡大する i_small = cv2.resize(i_rect, (size, size)) i_mos = cv2.resize(i_small, (w, h), interpolation=cv2.INTER_AREA) # 画像にモザイク画像を重ねる img2 = img.copy() img2[y1:y2, x1:x2] = i_mos return img2 class TestVideo(App): def build(self): return CameraClick() flgPlay = False flgFace = False flgMosaic = False TestVideo().run() |
ちょこっと解説
Windowサイズの指定
・1行目～3行目までで指定しております。
ファイルダイアログ
・82行目～99行目までで画面定義してます。FileChooserListViewを利用すると実現できます。
・ファイルを読み込むとテキストボックスにファイル名が表示されます。
動画再生
・再生ボタンをクリックするとテキストボックスに表示されているファイル名をcv2.VideoCaptureに引数として渡す事で再生します。
・もう一度再生ボタンをクリックすると動画は停止され、再度、再生すると最初から再生します。
最後に
cv2.VideoCapture の引数に動画ファイル名を指定するだけで簡単に改良できたのですが、Windowのサイズ設定やファイルダイアログの実装で試行錯誤してしまいました。
車やナンバーの認識は今回実装できておりません（顔認識のまま）ので、今後の課題です（＾＾;）
改良の余地ありありなので自分用試行錯誤メモな感じです。参考にされる場合はご注意ください。