【深度学习:目标检测】 py-faster-rcnn标注FDDB人脸便于其在FDDB上进行测试
来源:互联网 发布:红外透视镜软件下载 编辑:程序博客网 时间:2024/05/22 22:39
转载:http://blog.csdn.net/xzzppp/article/details/52071460
本程序是在py-faster-rcnn/tools/demo.py的基础上进行修改的
程序功能:用训练好的caffemodel,对FDDB人脸进行标注,便于其在FDDB上进行测试
- <span style="font-size:24px;">
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- import _init_paths
- from fast_rcnn.config import cfg
- from fast_rcnn.test import im_detect
- from fast_rcnn.nms_wrapper import nms
- from utils.timer import Timer
- import matplotlib.pyplot as plt
- import numpy as np
- import scipy.io as sio
- import caffe, os, sys, cv2
- import argparse
-
-
-
-
-
-
-
-
- CLASSES = ('__background__','face')
-
- NETS = {'vgg16': ('VGG16',
- 'VGG16_faster_rcnn_final.caffemodel'),
- 'myvgg': ('VGG_CNN_M_1024',
- 'VGG_CNN_M_1024_faster_rcnn_final.caffemodel'),
- 'zf': ('ZF',
- 'ZF_faster_rcnn_final.caffemodel'),
- 'myzf': ('ZF',
- 'zf_rpn_stage1_iter_80000.caffemodel'),
- }
-
-
- def vis_detections(im, class_name, dets, thresh=0.5):
-
- inds = np.where(dets[:, -1] >= thresh)[0]
- if len(inds) == 0:
- return
-
- write_file.write(str(len(inds)) + '\n')
- im = im[:, :, (2, 1, 0)]
-
-
- for i in inds:
- bbox = dets[i, :4]
- score = dets[i, -1]
-
-
- write_file.write( "{} {} {} {} {}\n".format(str(bbox[0]), str(bbox[1]),
- str(bbox[2] - bbox[0]),
- str(bbox[3] - bbox[1]),
- str(score)))
-
-
-
- ''
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- def demo(net, image_name):
-
-
-
-
- im = cv2.imread(image_name)
-
-
- timer = Timer()
- timer.tic()
- scores, boxes = im_detect(net, im)
- timer.toc()
- print ('Detection took {:.3f}s for '
- '{:d} object proposals').format(timer.total_time, boxes.shape[0])
-
-
- CONF_THRESH = 0.8
- NMS_THRESH = 0.3
- for cls_ind, cls in enumerate(CLASSES[1:]):
- cls_ind += 1
- cls_boxes = boxes[:, 4*cls_ind:4*(cls_ind + 1)]
- cls_scores = scores[:, cls_ind]
- dets = np.hstack((cls_boxes,
- cls_scores[:, np.newaxis])).astype(np.float32)
- keep = nms(dets, NMS_THRESH)
- dets = dets[keep, :]
- vis_detections(im, cls, dets, thresh=CONF_THRESH)
-
- def parse_args():
-
- parser = argparse.ArgumentParser(description='Faster R-CNN demo')
- parser.add_argument('--gpu', dest='gpu_id', help='GPU device id to use [0]',
- default=0, type=int)
- parser.add_argument('--cpu', dest='cpu_mode',
- help='Use CPU mode (overrides --gpu)',
- action='store_true')
- parser.add_argument('--net', dest='demo_net', help='Network to use [vgg16]',
- choices=NETS.keys(), default='vgg16')
-
- args = parser.parse_args()
-
- return args
-
- if __name__ == '__main__':
- cfg.TEST.HAS_RPN = True
-
- args = parse_args()
-
- prototxt = os.path.join(cfg.MODELS_DIR, NETS[args.demo_net][0],
- 'faster_rcnn_alt_opt', 'faster_rcnn_test.pt')
- caffemodel = os.path.join(cfg.DATA_DIR, 'faster_rcnn_models',
- NETS[args.demo_net][1])
-
- if not os.path.isfile(caffemodel):
- raise IOError(('{:s} not found.\nDid you run ./data/script/'
- 'fetch_faster_rcnn_models.sh?').format(caffemodel))
-
- if args.cpu_mode:
- caffe.set_mode_cpu()
- else:
- caffe.set_mode_gpu()
- caffe.set_device(args.gpu_id)
- cfg.GPU_ID = args.gpu_id
- net = caffe.Net(prototxt, caffemodel, caffe.TEST)
-
- print '\n\nLoaded network {:s}'.format(caffemodel)
-
-
- im = 128 * np.ones((300, 500, 3), dtype=np.uint8)
- for i in xrange(2):
- _, _= im_detect(net, im)
-
- ''
-
-
-
-
-
-
-
- for current_file in range(1,11):
-
- print 'Processing file ' + str(current_file) + ' ...'
-
- read_file_name = '/home/xiao/code/py-faster-rcnn-master/py-faster-rcnn/tools/FDDB-fold/FDDB-fold-' + str(current_file).zfill(2) + '.txt'
- write_file_name = '/home/xiao/code/py-faster-rcnn-master/py-faster-rcnn/tools/detections/fold-' + str(current_file).zfill(2) + '-out.txt'
- write_file = open(write_file_name, "w")
-
- with open(read_file_name, "r") as ins:
- array = []
- for line in ins:
- array.append(line)
-
- number_of_images = len(array)
-
- for current_image in range(number_of_images):
- if current_image % 10 == 0:
- print 'Processing image : ' + str(current_image)
-
- read_img_name = '/home/xiao/code/py-faster-rcnn-master/py-faster-rcnn/data/FDDB/originalPics/' + array[current_image].rstrip() + '.jpg'
- write_file.write(array[current_image])
- demo(net, read_img_name)
-
- write_file.close()
-
- ''
-
-
-
-
-
-
- plt.show()
- </span>
0 0