-
大小: 38.84MB文件類型: .zip金幣: 1下載: 0 次發布日期: 2023-08-08
- 語言: 其他
- 標簽: tensorflow??mtcnn??
資源簡介
tensorfolw訓練mtcnn模型 里面有詳細的指導文檔 教你一步步操作
代碼片段和文件信息
import?tensorflow?as?tf
import?numpy?as?np
class?Detector(object):
????#net_factory:rnet?or?onet
????#datasize:24?or?48
????def?__init__(self?net_factory?data_size?batch_size?model_path):
????????graph?=?tf.Graph()
????????with?graph.as_default():
????????????self.image_op?=?tf.placeholder(tf.float32?shape=[batch_size?data_size?data_size?3]?name=‘input_image‘)
????????????#figure?out?landmark????????????
????????????self.cls_prob?self.bbox_pred?self.landmark_pred?=?net_factory(self.image_op?training=False)
????????????self.sess?=?tf.Session(
????????????????config=tf.ConfigProto(allow_soft_placement=True?gpu_options=tf.GPUOptions(allow_growth=True)))
????????????saver?=?tf.train.Saver()
????????????#check?whether?the?dictionary?is?valid
????????????self.join?=?‘/‘.join(model_path.split(‘/‘)[:-1])
????????????model_dict?=?self.join
????????????ckpt?=?tf.train.get_checkpoint_state(model_dict)
????????????print(“model_path:?“?model_path)
????????????readstate?=?ckpt?and?ckpt.model_checkpoint_path
????????????assert??readstate?“the?params?dictionary?is?not?valid“
????????????print(“restore?models‘?param“)
????????????saver.restore(self.sess?model_path)
????????self.data_size?=?data_size
????????self.batch_size?=?batch_size
????#rnet?and?onet?minibatch(test)
????def?predict(self?databatch):
????????#?access?data
????????#?databatch:?N?x?3?x?data_size?x?data_size
????????scores?=?[]
????????batch_size?=?self.batch_size
????????minibatch?=?[]
????????cur?=?0
????????#num?of?all_data
????????n?=?databatch.shape[0]
????????while?cur?????????????#split?mini-batch
????????????minibatch.append(databatch[cur:min(cur?+?batch_size?n)?:?:?:])
????????????cur?+=?batch_size
????????#every?batch?prediction?result
????????cls_prob_list?=?[]
????????bbox_pred_list?=?[]
????????landmark_pred_list?=?[]
????????for?idx?data?in?enumerate(minibatch):
????????????m?=?data.shape[0]
????????????real_size?=?self.batch_size
????????????#the?last?batch?
????????????if?m?????????????????keep_inds?=?np.arange(m)
????????????????#gap?(difference)
????????????????gap?=?self.batch_size?-?m
????????????????while?gap?>=?len(keep_inds):
????????????????????gap?-=?len(keep_inds)
????????????????????keep_inds?=?np.concatenate((keep_inds?keep_inds))
????????????????if?gap?!=?0:
????????????????????keep_inds?=?np.concatenate((keep_inds?keep_inds[:gap]))
????????????????data?=?data[keep_inds]
????????????????real_size?=?m
????????????#cls_prob?batch*2
????????????#bbox_pred?batch*4
????????????cls_prob?bbox_predlandmark_pred?=?self.sess.run([self.cls_prob?self.bbox_predself.landmark_pred]?feed_dict={self.image_op:?data})
????????????#num_batch?*?batch_size?*2
????????????cls_prob_list.append(cls_prob[:real_size])
????????????#num_batch?*?batch_size?*4
????????????bbox_pred_list.append(bbox_pred[:real_size])
????????????#num_batch?*?batch_size*10
????????????landmark_pred_list.append(landmark_pred[:real_size])
????????????#num_of_data*2nu
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-07-10?16:12??MTCNN-Tensorflow-master\
?????目錄???????????0??2019-07-10?16:05??MTCNN-Tensorflow-master\.idea\
?????目錄???????????0??2019-06-11?10:06??MTCNN-Tensorflow-master\.idea\libraries\
?????文件?????????128??2019-06-11?10:06??MTCNN-Tensorflow-master\.idea\libraries\R_User_Library.xm
?????文件?????????185??2019-07-09?10:23??MTCNN-Tensorflow-master\.idea\misc.xm
?????文件?????????284??2018-10-16?02:08??MTCNN-Tensorflow-master\.idea\modules.xm
?????文件?????????576??2019-07-09?10:23??MTCNN-Tensorflow-master\.idea\MTCNN-Tensorflow.iml
?????文件???????31376??2019-07-10?16:05??MTCNN-Tensorflow-master\.idea\workspace.xm
?????目錄???????????0??2019-06-27?17:19??MTCNN-Tensorflow-master\data\
?????目錄???????????0??2018-10-16?02:08??MTCNN-Tensorflow-master\data\MTCNN_model\
?????目錄???????????0??2019-07-10?15:50??MTCNN-Tensorflow-master\data\MTCNN_model\ONet_landmark\
?????文件?????????187??2019-07-09?19:46??MTCNN-Tensorflow-master\data\MTCNN_model\ONet_landmark\checkpoint
?????文件?????3110276??2019-07-09?19:46??MTCNN-Tensorflow-master\data\MTCNN_model\ONet_landmark\ONet-22.data-00000-of-00001
?????文件????????1548??2019-07-09?19:46??MTCNN-Tensorflow-master\data\MTCNN_model\ONet_landmark\ONet-22.index
?????文件??????252409??2019-07-09?19:46??MTCNN-Tensorflow-master\data\MTCNN_model\ONet_landmark\ONet-22.me
?????目錄???????????0??2019-07-10?15:50??MTCNN-Tensorflow-master\data\MTCNN_model\PNet_landmark\
?????文件?????????187??2019-07-09?18:38??MTCNN-Tensorflow-master\data\MTCNN_model\PNet_landmark\checkpoint
?????文件???????55700??2019-07-09?18:38??MTCNN-Tensorflow-master\data\MTCNN_model\PNet_landmark\PNet-30.data-00000-of-00001
?????文件????????1185??2019-07-09?18:38??MTCNN-Tensorflow-master\data\MTCNN_model\PNet_landmark\PNet-30.index
?????文件??????209915??2019-07-09?18:38??MTCNN-Tensorflow-master\data\MTCNN_model\PNet_landmark\PNet-30.me
?????目錄???????????0??2019-07-10?15:51??MTCNN-Tensorflow-master\data\MTCNN_model\RNet_landmark\
?????文件?????????187??2019-07-09?18:08??MTCNN-Tensorflow-master\data\MTCNN_model\RNet_landmark\checkpoint
?????文件??????810724??2019-07-09?18:08??MTCNN-Tensorflow-master\data\MTCNN_model\RNet_landmark\RNet-22.data-00000-of-00001
?????文件????????1311??2019-07-09?18:08??MTCNN-Tensorflow-master\data\MTCNN_model\RNet_landmark\RNet-22.index
?????文件??????229545??2019-07-09?18:08??MTCNN-Tensorflow-master\data\MTCNN_model\RNet_landmark\RNet-22.me
?????目錄???????????0??2019-07-10?08:51??MTCNN-Tensorflow-master\Detection\
?????文件????????3164??2019-07-09?14:17??MTCNN-Tensorflow-master\Detection\detector.py
?????文件????????2509??2018-10-16?02:08??MTCNN-Tensorflow-master\Detection\detector.pyc
?????文件????????1854??2019-07-10?08:51??MTCNN-Tensorflow-master\Detection\fcn_detector.py
?????文件????????2047??2018-10-16?02:08??MTCNN-Tensorflow-master\Detection\fcn_detector.pyc
?????文件???????17999??2018-10-16?02:08??MTCNN-Tensorflow-master\Detection\MtcnnDetector.py
............此處省略144個文件信息
評論
共有 條評論