Exemplo n.º 1
0
    def initLabel(self):
        # Load common configuration
        (self.db_path, self.tb_name, self.imlist, self.labels) = \
            lc.load_cfg(lc.cfg_path)

        self.nimgs = len(self.imlist)
        self.npages = int(np.ceil(float(self.nimgs) / self.pagesize))

        # Check if the database exists
        lc.check_db(self.db_path, self.tb_name, self.imlist, self.labels)

        # Connect to the database
        self.conn = sqlite3.connect(self.db_path)
        self.cursor = self.conn.cursor()
Exemplo n.º 2
0
def main(argv):
    try:
        label1 = sys.argv[1]
        try:
            label2 = sys.argv[2]
        except:
            label_num = 1
    except:
        help(argv[0])
        sys.exit(1)

    # Load configuration
    (db_path, tb_name, imlist, labels) = lc.load_cfg(lc.cfg_path)
    (default_interval) = lc.load_cv_cfg(lc.cfg_path)

    # Check if the database exists
    lc.check_db(db_path, tb_name, imlist, labels)

    # Check if the labels are from the label list
    if (label1 not in labels.split()):
        print "Error: '%s' is not in the label list!" % label1
        lc.perr("\tSupported labels are: %s\n" % labels)
        sys.exit(1)

    if (2 == label_num and label2 not in labels.split()):
        print "Error: '%s' is not in the label list!" % label2
        lc.perr("\tSupported labels are: %s\n" % labels)
        sys.exit(1)

    # Adapt workint model
    print "Working mode: %d" % label_num
    # SQLs for update label information.
    a_ = 1048673
    s_ = 1048691
    d_ = 1048676
    f_ = 1048678

    update_sql = {}
    if (1 == label_num):
        print "\tLabel: %s" % label1
        # SQL for fetch labeled images
        prev_sql = "select img from %s where %s!='0'" % (tb_name, label1)
        # SQL for fetch unlabeled image
        fetch_sql = "select img, %s from %s where %s='0'" % (label1, tb_name,
                                                             label1)
        # SQL for view labeled image
        view_sql = "select img, %s from %s where img=?" % (label1, tb_name)
        # SQLs for update label information
        update_sql[d_] = "update %s set %s='t' where img=?" % (tb_name, label1)
        update_sql[f_] = "update %s set %s='f' where img=?" % (tb_name, label1)
        # Window Title for imshow
        imshow_title = label1
    elif (2 == label_num):
        print "\tLabel: %s %s" % (label1, label2)
        # SQL for fetch labeled images
        prev_sql = "select img from %s where %s!='0' and %s!='0'" % (
            tb_name, label1, label2)
        # SQL for fetch unlabeled image
        fetch_sql = "select img, %s, %s from %s where %s='0' or %s='0'" % (
            label1, label2, tb_name, label1, label2)
        # SQL for view labeled image
        view_sql = "select img, %s, %s from %s where img=?" % (label1, label2,
                                                               tb_name)
        # SQLs for update label information
        update_sql[a_] = "update %s set %s='t', %s='f' where img=?" % (
            tb_name, label1, label2)
        update_sql[s_] = "update %s set %s='f', %s='t' where img=?" % (
            tb_name, label1, label2)
        update_sql[d_] = "update %s set %s='t', %s='t' where img=?" % (
            tb_name, label1, label2)
        update_sql[f_] = "update %s set %s='f', %s='f' where img=?" % (
            tb_name, label1, label2)
        # Window Title for imshow
        imshow_title = "a: " + label1 + "    s: " + label2

    # Connect to the database
    conn = sqlite3.connect(db_path)
    c = conn.cursor()

    interval = 0
    proclst = []
    idx = -1

    c.execute(prev_sql)
    items = c.fetchall()
    proclst = [item[0] for item in items]
    idx = len(proclst) - 1

    prenum = len(proclst)

    c.execute(fetch_sql)
    item = c.fetchone()
    if (None == item):
        lc.perr("Labeling is finished, please choose other labels!\n")
        return
    proclst.append(item[0])
    idx += 1
    while (None != item):
        print idx, [str(x) for x in item]

        cv2.imshow(imshow_title, cv2.imread(item[0]))
        key = cv2.waitKey(interval)
        if (-1 == key):
            key = 1048678
        elif (key >= 1048624 and key <= 1048633):
            interval = 100 * (key - 1048624)
            continue

        # Esc pressed
        if (1048603 == key):
            conn.commit()
            break
        # j pressed, move forward
        elif (1048682 == key):
            interval = 0
            idx += 1
            print "up", idx
            if (idx >= len(proclst)):
                idx -= 1
                lc.perr("Last image!\n")
            else:
                c.execute(view_sql, (proclst[idx], ))
                item = c.fetchone()
            continue
        # k pressed, move backward
        elif (1048683 == key):
            interval = 0
            idx -= 1
            print "down", idx
            if (idx < 0):
                idx += 1
                lc.perr("First image!\n")
            else:
                c.execute(view_sql, (proclst[idx], ))
                item = c.fetchone()
            continue
        # a/s/d/f pressed
        elif (2 == label_num and key in [a_, s_, d_, f_]) or  \
             (1 == label_num and key in [d_, f_]):
            if (key == f_):
                interval = default_interval
            c.execute(update_sql[key], (item[0], ))
            # conn.commit();
            if (idx < len(proclst) - 1):
                idx += 1
                c.execute(view_sql, (proclst[idx], ))
                item = c.fetchone()
                continue
        else:
            print "Invalid op: %d" % key
            continue

        c.execute(fetch_sql)
        item = c.fetchone()
        if (None == item):
            break
        proclst.append(item[0])
        idx += 1

    print "You have labeled %d images this time!" % (len(proclst) - prenum)
    return
Exemplo n.º 3
0
def main(argv):
  try:
    label1 = sys.argv[1];
    try:
      label2 = sys.argv[2];
    except:
      label_num = 1;
  except:
    help(argv[0]);
    sys.exit(1);

  # Load configuration
  (db_path, tb_name, imlist, labels) = lc.load_cfg(lc.cfg_path);
  (default_interval) = lc.load_cv_cfg(lc.cfg_path);

  # Check if the database exists
  lc.check_db(db_path, tb_name, imlist, labels);

  # Check if the labels are from the label list
  if (label1 not in labels.split()):
    print "Error: '%s' is not in the label list!" % label1;
    lc.perr("\tSupported labels are: %s\n" % labels);
    sys.exit(1);

  if (2 == label_num and label2 not in labels.split()):
    print "Error: '%s' is not in the label list!" % label2;
    lc.perr("\tSupported labels are: %s\n" % labels);
    sys.exit(1);

  # Adapt workint model
  print "Working mode: %d" % label_num;
  # SQLs for update label information.
  a_ = 1048673; s_ = 1048691; d_ = 1048676; f_ = 1048678;
  
  update_sql = {};
  if (1 == label_num):
    print "\tLabel: %s" % label1;
    # SQL for fetch labeled images
    prev_sql  = "select img from %s where %s!='0'" % (tb_name, label1);
    # SQL for fetch unlabeled image
    fetch_sql = "select img, %s from %s where %s='0'" % (label1, tb_name, label1);
    # SQL for view labeled image
    view_sql  = "select img, %s from %s where img=?" % (label1, tb_name);
    # SQLs for update label information
    update_sql[d_] = "update %s set %s='t' where img=?" % (tb_name, label1);
    update_sql[f_] = "update %s set %s='f' where img=?" % (tb_name, label1);
    # Window Title for imshow
    imshow_title = label1;
  elif (2 == label_num):
    print "\tLabel: %s %s" % (label1, label2);
    # SQL for fetch labeled images
    prev_sql  = "select img from %s where %s!='0' and %s!='0'" % (tb_name, label1, label2);
    # SQL for fetch unlabeled image
    fetch_sql = "select img, %s, %s from %s where %s='0' or %s='0'" % (label1, label2, tb_name, label1, label2);
    # SQL for view labeled image
    view_sql  = "select img, %s, %s from %s where img=?" % (label1, label2, tb_name);
    # SQLs for update label information
    update_sql[a_] = "update %s set %s='t', %s='f' where img=?" % (tb_name, label1, label2);
    update_sql[s_] = "update %s set %s='f', %s='t' where img=?" % (tb_name, label1, label2);
    update_sql[d_] = "update %s set %s='t', %s='t' where img=?" % (tb_name, label1, label2);
    update_sql[f_] = "update %s set %s='f', %s='f' where img=?" % (tb_name, label1, label2);
    # Window Title for imshow
    imshow_title = "a: " + label1 + "    s: " + label2;

  # Connect to the database
  conn = sqlite3.connect(db_path)
  c = conn.cursor();

  interval = 0;
  proclst = [];
  idx = -1;

  c.execute(prev_sql);
  items = c.fetchall();
  proclst = [item[0] for item in items];
  idx = len(proclst) - 1;
  
  prenum = len(proclst);

  c.execute(fetch_sql);
  item = c.fetchone();
  if (None  == item):
    lc.perr("Labeling is finished, please choose other labels!\n");
    return;
  proclst.append(item[0]); idx += 1;
  while (None != item):
    print idx, [str(x) for x in item];

    cv2.imshow(imshow_title, cv2.imread(item[0]));
    key = cv2.waitKey(interval);
    if (-1 == key):
      key = 1048678;
    elif (key >= 1048624 and key <= 1048633):
      interval = 100 * (key - 1048624);
      continue;
      

    # Esc pressed
    if (1048603 == key):
      conn.commit();
      break;
    # j pressed, move forward
    elif (1048682 == key):
      interval = 0;
      idx += 1;
      print "up", idx;
      if (idx >= len(proclst)):
        idx -= 1;
        lc.perr("Last image!\n");
      else:
        c.execute(view_sql, (proclst[idx], ));
        item = c.fetchone();
      continue;
    # k pressed, move backward
    elif (1048683 == key):
      interval = 0;
      idx -= 1;
      print "down", idx;
      if (idx < 0):
        idx += 1;
        lc.perr("First image!\n");
      else:
        c.execute(view_sql, (proclst[idx], ));
        item = c.fetchone();
      continue;
    # a/s/d/f pressed
    elif (2 == label_num and key in [a_, s_, d_, f_]) or  \
         (1 == label_num and key in [d_, f_]):
      if (key == f_):
        interval = default_interval;
      c.execute(update_sql[key],  (item[0], ));
      # conn.commit();
      if (idx < len(proclst) - 1):
        idx += 1;
        c.execute(view_sql, (proclst[idx], ));
        item = c.fetchone();
        continue;
    else:
      print "Invalid op: %d" % key;
      continue;

    c.execute(fetch_sql);
    item = c.fetchone();
    if (None == item):
      break;
    proclst.append(item[0]); idx += 1;

  print "You have labeled %d images this time!" % (len(proclst) - prenum)
  return;