first init

Author: taokongcn

data/ImageNet_models/dump

@@ -0,0 +1 @@
+init

data/RON_models/dump

@@ -0,0 +1 @@
+init

demo_camera.sh

@@ -0,0 +1 @@
+python tools/demo_video.py

lib/Makefile

@@ -0,0 +1,6 @@
+all:
+	python setup.py build_ext --inplace
+	rm -rf build
+clean:
+	rm -rf nms/*.so nms/*.c
+	rm -rf utils/*.so utils/*.c

lib/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m

@@ -0,0 +1,14 @@
+function VOCopts = get_voc_opts(path)
+
+tmp = pwd;
+cd(path);
+try
+  addpath('VOCcode');
+  VOCinit;
+catch
+  rmpath('VOCcode');
+  cd(tmp);
+  error(sprintf('VOCcode directory not found under %s', path));
+end
+rmpath('VOCcode');
+cd(tmp);

lib/datasets/VOCdevkit-matlab-wrapper/voc_eval.m

@@ -0,0 +1,60 @@
+function res = voc_eval(path, comp_id, test_set, output_dir, rm_res)
+
+VOCopts = get_voc_opts(path);
+VOCopts.testset = test_set;
+
+for i = 1:length(VOCopts.classes)
+  cls = VOCopts.classes{i};
+  res(i) = voc_eval_cls(cls, VOCopts, comp_id, output_dir, rm_res);
+end
+
+fprintf('\n~~~~~~~~~~~~~~~~~~~~\n');
+fprintf('Results:\n');
+aps = [res(:).ap]';
+fprintf('%.1f\n', aps * 100);
+fprintf('%.1f\n', mean(aps) * 100);
+fprintf('~~~~~~~~~~~~~~~~~~~~\n');
+
+function res = voc_eval_cls(cls, VOCopts, comp_id, output_dir, rm_res)
+
+test_set = VOCopts.testset;
+year = VOCopts.dataset(4:end);
+
+addpath(fullfile(VOCopts.datadir, 'VOCcode'));
+
+res_fn = sprintf(VOCopts.detrespath, comp_id, cls);
+
+recall = [];
+prec = [];
+ap = 0;
+ap_auc = 0;
+
+do_eval = (str2num(year) <= 2007) | ~strcmp(test_set, 'test');
+if do_eval
+  % Bug in VOCevaldet requires that tic has been called first
+  tic;
+  [recall, prec, ap] = VOCevaldet(VOCopts, comp_id, cls, true);
+  ap_auc = xVOCap(recall, prec);
+
+  % force plot limits
+  ylim([0 1]);
+  xlim([0 1]);
+
+  print(gcf, '-djpeg', '-r0', ...
+        [output_dir '/' cls '_pr.jpg']);
+end
+fprintf('!!! %s : %.4f %.4f\n', cls, ap, ap_auc);
+
+res.recall = recall;
+res.prec = prec;
+res.ap = ap;
+res.ap_auc = ap_auc;
+
+save([output_dir '/' cls '_pr.mat'], ...
+     'res', 'recall', 'prec', 'ap', 'ap_auc');
+
+if rm_res
+  delete(res_fn);
+end
+
+rmpath(fullfile(VOCopts.datadir, 'VOCcode'));

lib/datasets/VOCdevkit-matlab-wrapper/xVOCap.m

@@ -0,0 +1,10 @@
+function ap = xVOCap(rec,prec)
+% From the PASCAL VOC 2011 devkit
+
+mrec=[0 ; rec ; 1];
+mpre=[0 ; prec ; 0];
+for i=numel(mpre)-1:-1:1
+    mpre(i)=max(mpre(i),mpre(i+1));
+end
+i=find(mrec(2:end)~=mrec(1:end-1))+1;
+ap=sum((mrec(i)-mrec(i-1)).*mpre(i));

lib/datasets/__init__.py

@@ -0,0 +1,43 @@
+# --------------------------------------------------------
+# Fast R-CNN
+# Copyright (c) 2015 Microsoft
+# Licensed under The MIT License [see LICENSE for details]
+# Written by Ross Girshick
+# --------------------------------------------------------
+
+from .imdb import imdb
+from .pascal_voc import pascal_voc
+from .coco import coco
+from . import factory
+
+import os.path as osp
+ROOT_DIR = osp.join(osp.dirname(__file__), '..', '..')
+
+# We assume your matlab binary is in your path and called `matlab'.
+# If either is not true, just add it to your path and alias it as matlab, or
+# you could change this file.
+MATLAB = 'matlab'
+
+# http://stackoverflow.com/questions/377017/test-if-executable-exists-in-python
+def _which(program):
+    import os
+    def is_exe(fpath):
+        return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
+
+    fpath, fname = os.path.split(program)
+    if fpath:
+        if is_exe(program):
+            return program
+    else:
+        for path in os.environ["PATH"].split(os.pathsep):
+            path = path.strip('"')
+            exe_file = os.path.join(path, program)
+            if is_exe(exe_file):
+                return exe_file
+
+    return None
+
+if _which(MATLAB) is None:
+    msg = ("MATLAB command '{}' not found. "
+           "Please add '{}' to your PATH.").format(MATLAB, MATLAB)
+    #raise EnvironmentError(msg)