mask-train.py by yolov5

ようやく、yolov5のtrain.pyでmask-detect.pyをすことができた。結局処理能力の問題か?!

YOLOv5 PytorchでMask Wearing Datasetを使用。先日のJetsonNanoでは、yolov5s-20epochesまでしか動かず、マスク検出もできない結果であった。

で、hp-pc_i5第7世代/GPUなしubuntu20.04(torch1.6/torchvision0.7)で、trainを実施。yolovs5s-2000epochesで、

f:id:vanlaborg:20210119003558p:plain

22時間のtrain.py


train.py22時間廻しっぱなし。そしてjetsonNanoのdetect.py結果、ようやくマスク検出ができた。

f:id:vanlaborg:20210119003657p:plain

maskと定番のジダン