Browse Source

print function

fetches/miaozaiye/master
kkkim 7 years ago
parent
commit
87f36fc8e9
16 changed files with 18 additions and 16 deletions
  1. +1
    -1
      dface/core/detect.py
  2. +1
    -0
      dface/core/imagedb.py
  3. +1
    -1
      dface/prepare_data/assemble_onet_imglist.py
  4. +1
    -1
      dface/prepare_data/assemble_pnet_imglist.py
  5. +1
    -1
      dface/prepare_data/assemble_rnet_imglist.py
  6. +2
    -1
      dface/prepare_data/gen_Onet_train_data.py
  7. +1
    -0
      dface/prepare_data/gen_Pnet_train_data.py
  8. +1
    -2
      dface/prepare_data/gen_Rnet_train_data.py
  9. +1
    -1
      dface/prepare_data/gen_landmark_12.py
  10. +1
    -1
      dface/prepare_data/gen_landmark_24.py
  11. +1
    -1
      dface/prepare_data/gen_landmark_48.py
  12. +2
    -0
      dface/prepare_data/gen_landmark_net_48.py
  13. +1
    -3
      dface/train_net/train.py
  14. +1
    -1
      dface/train_net/train_o_net.py
  15. +1
    -1
      dface/train_net/train_p_net.py
  16. +1
    -1
      dface/train_net/train_r_net.py

+ 1
- 1
dface/core/detect.py View File

@@ -6,7 +6,7 @@ from torch.autograd.variable import Variable
from dface.core.models import PNet,RNet,ONet
import dface.core.utils as utils
import dface.core.image_tools as image_tools
from __future__ import print_function

def create_mtcnn_net(p_model_path=None, r_model_path=None, o_model_path=None, use_cuda=True):



+ 1
- 0
dface/core/imagedb.py View File

@@ -1,5 +1,6 @@
import os
import numpy as np
from __future__ import print_function

class ImageDB(object):
def __init__(self, image_annotation_file, prefix_path='', mode='train'):


+ 1
- 1
dface/prepare_data/assemble_onet_imglist.py View File

@@ -1,7 +1,7 @@
import os
import dface.config as config
import dface.prepare_data.assemble as assemble
from __future__ import print_function

if __name__ == '__main__':



+ 1
- 1
dface/prepare_data/assemble_pnet_imglist.py View File

@@ -1,7 +1,7 @@
import os
import dface.config as config
import dface.prepare_data.assemble as assemble
from __future__ import print_function

if __name__ == '__main__':



+ 1
- 1
dface/prepare_data/assemble_rnet_imglist.py View File

@@ -1,7 +1,7 @@
import os
import dface.config as config
import dface.prepare_data.assemble as assemble
from __future__ import print_function

if __name__ == '__main__':



+ 2
- 1
dface/prepare_data/gen_Onet_train_data.py View File

@@ -1,5 +1,4 @@
import argparse

import cv2
import numpy as np
from dface.core.detect import MtcnnDetector,create_mtcnn_net
@@ -11,6 +10,8 @@ import cPickle
from dface.core.utils import convert_to_square,IoU
import dface.config as config
import dface.core.vision as vision
from __future__ import print_function


def gen_onet_data(data_dir, anno_file, pnet_model_file, rnet_model_file, prefix_path='', use_cuda=True, vis=False):



+ 1
- 0
dface/prepare_data/gen_Pnet_train_data.py View File

@@ -5,6 +5,7 @@ import os
import numpy.random as npr
from dface.core.utils import IoU
import dface.config as config
from __future__ import print_function

def gen_pnet_data(data_dir,anno_file,prefix):



+ 1
- 2
dface/prepare_data/gen_Rnet_train_data.py View File

@@ -1,6 +1,4 @@
import argparse


import cv2
import numpy as np
from dface.core.detect import MtcnnDetector,create_mtcnn_net
@@ -12,6 +10,7 @@ import cPickle
from dface.core.utils import convert_to_square,IoU
import dface.config as config
import dface.core.vision as vision
from __future__ import print_function

def gen_rnet_data(data_dir, anno_file, pnet_model_file, prefix_path='', use_cuda=True, vis=False):



+ 1
- 1
dface/prepare_data/gen_landmark_12.py View File

@@ -7,7 +7,7 @@ import numpy.random as npr
import argparse
import dface.config as config
import dface.core.utils as utils
from __future__ import print_function

def gen_data(anno_file, data_dir, prefix):



+ 1
- 1
dface/prepare_data/gen_landmark_24.py View File

@@ -8,7 +8,7 @@ import numpy.random as npr
import argparse
import dface.config as config
import dface.core.utils as utils
from __future__ import print_function


def gen_data(anno_file, data_dir, prefix):


+ 1
- 1
dface/prepare_data/gen_landmark_48.py View File

@@ -8,7 +8,7 @@ import numpy.random as npr
import argparse
import dface.config as config
import dface.core.utils as utils
from __future__ import print_function

def gen_data(anno_file, data_dir, prefix):



+ 2
- 0
dface/prepare_data/gen_landmark_net_48.py View File

@@ -11,6 +11,8 @@ import cPickle
from dface.core.utils import convert_to_square,IoU
import dface.config as config
import dface.core.vision as vision
from __future__ import print_function


def gen_landmark48_data(data_dir, anno_file, pnet_model_file, rnet_model_file, prefix_path='', use_cuda=True, vis=False):



+ 1
- 3
dface/train_net/train.py View File

@@ -5,9 +5,7 @@ from dface.core.models import PNet,RNet,ONet,LossFn
import torch
from torch.autograd import Variable
import dface.core.image_tools as image_tools



from __future__ import print_function


def compute_accuracy(prob_cls, gt_cls):


+ 1
- 1
dface/train_net/train_o_net.py View File

@@ -4,7 +4,7 @@ from dface.core.imagedb import ImageDB
import dface.train_net.train as train
import dface.config as config
import os
from __future__ import print_function


def train_net(annotation_file, model_store_path,


+ 1
- 1
dface/train_net/train_p_net.py View File

@@ -4,7 +4,7 @@ from dface.core.imagedb import ImageDB
from dface.train_net.train import train_pnet
import dface.config as config
import os
from __future__ import print_function


def train_net(annotation_file, model_store_path,


+ 1
- 1
dface/train_net/train_r_net.py View File

@@ -4,7 +4,7 @@ from dface.core.imagedb import ImageDB
import dface.train_net.train as train
import dface.config as config
import os
from __future__ import print_function


def train_net(annotation_file, model_store_path,


Loading…
Cancel
Save