質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.53%
Google Colaboratory

Google Colaboratoryとは、無償のJupyterノートブック環境。教育や研究機関の機械学習の普及のためのGoogleの研究プロジェクトです。PythonやNumpyといった機械学習で要する大方の環境がすでに構築されており、コードの記述・実行、解析の保存・共有などが可能です。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

1回答

1844閲覧

抽出する画像の枚数を制限したい。

Keiti

総合スコア5

Google Colaboratory

Google Colaboratoryとは、無償のJupyterノートブック環境。教育や研究機関の機械学習の普及のためのGoogleの研究プロジェクトです。PythonやNumpyといった機械学習で要する大方の環境がすでに構築されており、コードの記述・実行、解析の保存・共有などが可能です。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2020/07/18 09:08

編集2020/07/19 09:36

以下のコードは動画を読み込み、その動画内に写ったマスクをつけた人物の顔を切り取り、画像として抽出するコードになります。
実行自体は上手く行くのですが、動画のフレーム数を全て画像として抽出してしまうので、長い動画を読み込んだ際には必要以上に画像が生成されてしまいます(例えば、1000のフレームで構成された動画を読み込むと必要な写真は100枚にもかかわらず1000枚生成されてしまいます)。
そこで、生成する画像の枚数を制限したいのですがどのようにコードを書き換えればよいかわかりません。どうかご教示ください。

from

1drive.mount('/content/drive') 2from google.colab.patches import cv2_imshow 3import paddlehub as hub 4import cv2 5from PIL import Image, ImageDraw, ImageFont 6import numpy as np 7import json 8import os 9 10module = hub.Module(name="pyramidbox_lite_server_mask", version='1.1.0') 11 12def paint_chinese(im, chinese, position, fontsize, color_bgr): 13 img_PIL = Image.fromarray(cv2.cvtColor(im, cv2.COLOR_BGR2RGB)) 14 font = ImageFont.truetype( 15 'SourceHanSansSC-Medium.otf', fontsize, encoding="utf-8") 16 color = color_bgr[::-1] 17 draw = ImageDraw.Draw(img_PIL) 18 draw.text(position, chinese, font=font, fill=color) 19 img = cv2.cvtColor(np.asarray(img_PIL), cv2.COLOR_RGB2BGR) 20 return img 21 22 23result_path = './result' 24if not os.path.exists(result_path): 25 os.mkdir(result_path) 26 27name = "./result/1-mask_detection.mp4" 28width = 1280 29height = 720 30fps = 30 31fourcc = cv2.VideoWriter_fourcc('m','p','4', 'v') # changed *'vp90' -> 'm','p','4', 'v' 32writer = cv2.VideoWriter(name, fourcc, fps, (width, height)) 33 34maskIndex = 0 35index = 0 36data = [] 37 38capture = cv2.VideoCapture('/content/drive/My Drive/sozai/5.mp4') 39 40while True: 41 frameData = {} 42 ret, frame = capture.read() 43 if ret == False: 44 break 45 46 frame_copy = frame.copy() 47 input_dict = {"data": [frame]} 48 results = module.face_detection(data=input_dict) 49 50 maskFrameDatas = [] 51 for result in results: 52 label = result['data']['label'] 53 confidence_origin = result['data']['confidence'] 54 confidence = round(confidence_origin, 2) 55 confidence_desc = str(confidence) 56 57 top, right, bottom, left = int(result['data']['top']), int( 58 result['data']['right']), int(result['data']['bottom']), int( 59 result['data']['left']) 60 61 img_name = "avatar_%d.png" % (maskIndex) 62 path = "/content/drive/My Drive/Face/A_face/" + img_name 63 image = frame[top - 10:bottom + 10, left - 10:right + 10] 64 cv2.imwrite(path, image, [int(cv2.IMWRITE_PNG_COMPRESSION), 9]) 65 66 maskFrameData = {} 67 maskFrameData['top'] = top 68 maskFrameData['right'] = right 69 maskFrameData['bottom'] = bottom 70 maskFrameData['left'] = left 71 maskFrameData['confidence'] = float(confidence_origin) 72 maskFrameData['label'] = label 73 maskFrameData['img'] = img_name 74 75 maskFrameDatas.append(maskFrameData) 76 77 maskIndex += 1 78 79 color = (0, 255, 0) 80 label_cn = "有口罩" 81 if label == 'NO MASK': 82 color = (0, 0, 255) 83 label_cn = "无口罩" 84 85 cv2.rectangle(frame_copy, (left, top), (right, bottom), color, 3) 86 cv2.putText(frame_copy, label, (left, top-10), cv2.FONT_HERSHEY_SIMPLEX, 0.8, color, 2) 87 88 writer.write(frame_copy) 89 90 frameData['frame'] = index 91 frameData['data'] = maskFrameDatas 92 93 data.append(frameData) 94 print(json.dumps(frameData)) 95 96 index += 1 97 98 if cv2.waitKey(1) & 0xFF == ord('q'): 99 break 100 101with open("./result/2-mask_detection.json", "w") as f: 102 json.dump(data, f) 103 104writer.release() 105 106cv2.destroyAllWindows() 107

個人的にチャレンジしてみたことはindex,maskindex,fpsなどの値を変えてみたことです。frameData['frame']=index
などと書かれているので、indexの値を制限すれば取得するframe数も制限されると考えました。
ちなみにこのコードは以下のサイトから引用させていただきました。
https://qiita.com/ebiflyyyyyyyy/items/4f8cc781a86c3f9d8700)

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

aokikenichi

2020/07/18 09:17

Pythonなのでインデントなしだと読めないです インデントをお願いいたします
meg_

2020/07/18 13:06

質問のコードはどこのものでしょうか?
can110

2020/07/19 10:05 編集

各フレームにおいて画像(.png)を保存すべきかどうかはどのようなルールで決めるのでしょうか? 無条件に先頭の100フレームでよいでしょうか? あるいはマスクをつけた画像のみ保存ということでしょうか?
Keiti

2020/07/19 10:32

実は欲を言えば無作為に保存したいんですよね・・・例えば1,34,56,78,,,,みたいな感じが理想的です!
can110

2020/07/19 10:37

ソースコード上で画像ファイルを保存しているところ(行)はどこか分かりますでしょうか?
guest

回答1

0

ベストアンサー

画像ファイル保存処理の部分を以下のようなif文で囲んで保存可否を切り替えればよいかと思います。

Python

1import random 23 if random.random() > 0.9: # 例えばこんな感じで無作為に 4 img_name = "avatar_%d.png" % (maskIndex) 5 path = "/content/drive/My Drive/Face/A_face/" + img_name 6 image = frame[top - 10:bottom + 10, left - 10:right + 10] 7 cv2.imwrite(path, image, [int(cv2.IMWRITE_PNG_COMPRESSION), 9]) 8 9 maskIndex += 1

投稿2020/07/19 11:05

can110

総合スコア38228

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

Keiti

2020/07/20 09:29

遅くなってすみません!できました!ありがとうございます!
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.53%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問