blob: bfb8563128655ed6926a70762ddfa696bf9c4c69 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
from utils.dataset import CASIAB, ClipConditions, ClipViews
CASIAB_ROOT_DIR = '../data/CASIA-B-MRCNN/SEG'
def test_casiab():
casiab = CASIAB(CASIAB_ROOT_DIR, discard_threshold=0)
assert len(casiab) == 74 * 10 * 11
def test_casiab_nm():
nm_selector = {'conditions': ClipConditions({r'nm-0\d'})}
casiab_nm = CASIAB(CASIAB_ROOT_DIR, selector=nm_selector,
discard_threshold=0)
assert len(casiab_nm) == 74 * 6 * 11
def test_casiab_nm_bg_90():
nm_bg_90_selector = {'conditions': ClipConditions({r'nm-0\d', r'bg-0\d'}),
'views': ClipViews({'090'})}
casiab_nm_bg_90 = CASIAB(CASIAB_ROOT_DIR,
selector=nm_bg_90_selector,
discard_threshold=0)
assert len(casiab_nm_bg_90) == 74 * (6 + 2) * 1
|
