Upgrade to Pro
— share decks privately, control downloads, hide ads and more …
Speaker Deck
Speaker Deck
PRO
Sign in
Sign up
for free
ディープラーニング素人がPyTorchでマスクつけている人を学習させてみた
KMiura
February 20, 2021
Technology
0
360
ディープラーニング素人がPyTorchでマスクつけている人を学習させてみた
2021.02.20「Python Charity Talks in Japan 2021.02」
https://pyconjp.connpass.com/event/199787/
KMiura
February 20, 2021
Tweet
Share
More Decks by KMiura
See All by KMiura
miura55
0
280
miura55
0
26
miura55
1
280
miura55
0
120
miura55
0
210
miura55
0
250
miura55
0
200
miura55
0
210
miura55
0
110
Other Decks in Technology
See All in Technology
sat
40
29k
imdigitallab
0
520
harshbothra
0
150
kahara33
0
120
hiroyaiizuka
0
140
kkosukeee
0
160
mahito
0
230
dena_tech
1
670
minma
0
210
yuji1484
3
850
youtalk
0
430
koukyo1994
3
540
Featured
See All Featured
malarkey
392
61k
myddelton
109
11k
pauljervisheath
195
15k
sferik
610
55k
bryan
31
3.4k
dougneiner
118
7.9k
mthomps
38
2.3k
deanohume
294
28k
geoffreycrofte
21
930
bkeepers
408
58k
andyhume
63
3.7k
erikaheidi
14
4.3k
Transcript
ディープラーニング素人がPyTorchで マスクつけている人を学習させてみた KMiura(@k_miura_io)
お前だれよ • 三浦 耕生(こうき) • 名古屋のゲーム会社 のサーバーエンジニア @k_miura_io koki.miura05
ディープラーニング • ディープラーニングはチュートリアル 止まり • せっかくGPU乗ったSurface買ったの に…(3年前) • Pythonやってるならディープラーニ ングやらないと
ディープラーニングを社会の役に立てたい (建前) • 外出するのにマスクを付けるのは当 たり前 • 百貨店行くとマスクつけても認識する サーモカメラを見かけるようになった • ワイも実装してドヤりたい(本音)
実行環境 • Python 3.9 • Pytorch(Tensorflowはベタすぎる) • NVIDIA GeForce GTX
965M(Surface Book内蔵GPU)
とりあえず動かしてみた • Mobile Net V2を転移学習 • データ数600枚ほど • 思ってたのと違う…
Why? • 逆伝搬の計算するとトレーニングをするたびに以前の勾配を加算する • トレーニングの度に勾配をゼロにする処理をいれてみた • ついでにデータも増やしたれ(各クラス600枚→900枚)
学習結果 • 悪くはないがノイズ結構のってる • Accuracyが1に到達してるものもある(収束しているのが理想)
さらに改善 • もっとデータ量を増やしてみた(各クラス900枚→1600枚) • ついでにエポック数増やしてみる(100エポック→150エポック)
学習結果 • 目立ったノイズも無くきれいな曲線(ディープラーニングでよくあるやつ) • Accuracyが1に到達せず収束している • やはりデータがものをいう
実験成果 https://youtu.be/VbBSG1Is3hY
まとめ • ディープラーニングはデータ量、アルゴリズムは二の次 • ようやくGPUを生かすことができた(Jetson Nanoも買える) • ディープラーニング完全に理解した
今回のソースコード&ブログ https://github.com/Miura55/deeplearning- face-mask-detection https://supernove.hatenadiary.jp/entry/ 2021/01/19/041445
END