LeslieZhoa / tensorflow-MTCNN

人脸检测MTCNN算法,采用tensorflow框架编写,从理解到训练,中文注释完全,含测试和训练,支持摄像头
364 stars 141 forks source link

train_model代码中net_factory含义应该是什么,应该是调用train目录下的model代码块,但是没有看到import该目录 #14

Open bestlyh opened 5 years ago

jt-z commented 5 years ago

实际上train_model.py中的train函数仅仅是提供了一个通用的对三种网络的训练流程,其中的net_factory参数代表三种不同的网络,会在train函数被调用时由调用者给出Pnet、Rnet或者ONet的网络实例。
具体训练时是在train.py这个文件中实现的训练,这个文件中导入了train_model中的train函数,train_model.py只是提供了一个类似于工具类的部分。