mode=0: only body; mode=1: only head
(self, gtpath=None, dtpath=None, body_key=None, head_key=None, mode=0)
| 483 | |
| 484 | class Database(object): |
| 485 | def __init__(self, gtpath=None, dtpath=None, body_key=None, head_key=None, mode=0): |
| 486 | """ |
| 487 | mode=0: only body; mode=1: only head |
| 488 | """ |
| 489 | self.images = dict() |
| 490 | self.eval_mode = mode |
| 491 | self.loadData(gtpath, body_key, head_key, if_gt=True) |
| 492 | self.loadData(dtpath, body_key, head_key, if_gt=False) |
| 493 | |
| 494 | self._ignNum = sum([self.images[i]._ignNum for i in self.images]) |
| 495 | self._gtNum = sum([self.images[i]._gtNum for i in self.images]) |
| 496 | self._imageNum = len(self.images) |
| 497 | self.scorelist = None |
| 498 | |
| 499 | def loadData(self, fpath, body_key=None, head_key=None, if_gt=True): |
| 500 | # assert os.path.isfile(fpath), fpath + " does not exist!" |