資源簡(jiǎn)介
**DFace** 是個(gè)開源的深度學(xué)習(xí)人臉檢測(cè)和人臉識(shí)別系統(tǒng)。所有功能都采用 **[pytorch](https://github.com/pytorch/pytorch)** 框架開發(fā)。pytorch是一個(gè)由facebook開發(fā)的深度學(xué)習(xí)框架，它包含了一些比較有趣的高級(jí)特性，例如自動(dòng)求導(dǎo)，動(dòng)態(tài)構(gòu)圖等。DFace天然的繼承了這些優(yōu)點(diǎn)，使得它的訓(xùn)練過(guò)程可以更加簡(jiǎn)單方便，并且實(shí)現(xiàn)的代碼可以更加清晰易懂。
DFace可以利用CUDA來(lái)支持GPU加速模式。我們建議嘗試linux GPU這種模式，它幾乎可以實(shí)現(xiàn)實(shí)時(shí)的效果。
代碼片段和文件信息
import?cv2
from?dface.core.detect?import?create_mtcnn_net?MtcnnDetector
from?dface.core?import?vision
if?__name__?==?‘__main__‘:
????#?refer?to?your?local?model?path?
????p_model?=?“./model_store/pnet_epoch.pt“
????r_model?=?“./model_store/rnet_epoch.pt“
????o_model?=?“./model_store/onet_epoch.pt“
????#use?cpu?version?set?use_cuda=False?if?you?want?to?use?gpu?version?set?use_cuda=True
????pnet?rnet?onet?=?create_mtcnn_net（p_model_path=p_model?r_model_path=r_model?o_model_path=o_model?use_cuda=False）
????mtcnn_detector?=?MtcnnDetector（pnet=pnet?rnet=rnet?onet=onet?min_face_size=24）
????img?=?cv2.imread（“./test.jpg“）
????b?g?r?=?cv2.split（img）
????img2?=?cv2.merge（[r?g?b]）
????bboxs?landmarks?=?mtcnn_detector.detect_face（img）
????#?print?box_align
????vision.vis_face（img2bboxslandmarks）
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-03-28?17:56??dface\
?????文件???????10250??2018-03-28?17:56??dface\LICENSE
?????文件????????8055??2018-03-28?17:56??dface\README.md
?????目錄???????????0??2018-03-28?17:56??dface\anno_store\
?????文件???????????0??2018-03-28?17:56??dface\anno_store\__init__.py
?????文件??????????55??2018-03-28?17:56??dface\anno_store\info
?????文件?????3208713??2018-03-28?17:56??dface\anno_store\wider_origin_anno.txt
?????目錄???????????0??2018-03-28?17:56??dface\dface\
?????文件???????????0??2018-03-28?17:56??dface\dface\__init__.py
?????文件????????1037??2018-03-28?17:56??dface\dface\config.py
?????目錄???????????0??2018-03-28?17:56??dface\dface\core\
?????文件???????????0??2018-03-28?17:56??dface\dface\core\__init__.py
?????文件???????22527??2018-03-28?17:56??dface\dface\core\detect.py
?????文件????????4224??2018-03-28?17:56??dface\dface\core\image_reader.py
?????文件????????1180??2018-03-28?17:56??dface\dface\core\image_tools.py
?????文件????????5310??2018-03-28?17:56??dface\dface\core\imagedb.py
?????文件????????6848??2018-03-28?17:56??dface\dface\core\models.py
?????文件????????1150??2018-03-28?17:56??dface\dface\core\nms.py
?????文件????????8950??2018-03-28?17:56??dface\dface\core\resnet_inception_v2.py
?????文件??????????20??2018-03-28?17:56??dface\dface\core\roc.py
?????文件????????2653??2018-03-28?17:56??dface\dface\core\utils.py
?????文件????????5136??2018-03-28?17:56??dface\dface\core\vision.py
?????目錄???????????0??2018-03-28?17:56??dface\dface\prepare_data\
?????文件???????????0??2018-03-28?17:56??dface\dface\prepare_data\__init__.py
?????文件?????????979??2018-03-28?17:56??dface\dface\prepare_data\assemble.py
?????文件?????????984??2018-03-28?17:56??dface\dface\prepare_data\assemble_onet_imglist.py
?????文件?????????996??2018-03-28?17:56??dface\dface\prepare_data\assemble_pnet_imglist.py
?????文件?????????996??2018-03-28?17:56??dface\dface\prepare_data\assemble_rnet_imglist.py
?????文件????????8007??2018-03-28?17:56??dface\dface\prepare_data\gen_Onet_train_data.py
?????文件????????6709??2018-03-28?17:56??dface\dface\prepare_data\gen_Pnet_train_data.py
?????文件????????7696??2018-03-28?17:56??dface\dface\prepare_data\gen_Rnet_train_data.py
............此處省略30個(gè)文件信息
評(píng)論
共有 條評(píng)論
川公網(wǎng)安備 51152502000135號(hào)