# Label('bus', rgb2bgr((230, 150, 140))),
# Label('car', rgb2bgr(( 70, 70, 70))),
# Label('cat', rgb2bgr((102, 102, 156))),
# Label('chair', rgb2bgr((190, 153, 153))),
# Label('cow', rgb2bgr((150, 120, 90))),
# Label('diningtable', rgb2bgr((153, 153, 153))),
# Label('dog', rgb2bgr((250, 170, 30))),
# Label('horse', rgb2bgr((220, 220, 0))),
# Label('motorbike', rgb2bgr((107, 142, 35))),
# Label('person', rgb2bgr(( 52, 151, 52)))
# Label('pottedplant', rgb2bgr(( 70, 130, 180))),
# Label('sheep', rgb2bgr((220, 20, 60))),
# Label('sofa', rgb2bgr(( 0, 0, 142))),
# Label('train', rgb2bgr(( 0, 0, 230))),
# Label('tvmonitor', rgb2bgr((119, 11, 32)))
Label('face', rgb2bgr((255,0,0)))
]
#-------------------------------------------------------------------------------
class PascalVOCSource:
#---------------------------------------------------------------------------
def __init__(self):
self.num_classes = len(label_defs)
self.colors = {l.name: l.color for l in label_defs}
self.lid2name = {i: l.name for i, l in enumerate(label_defs)}
self.lname2id = {l.name: i for i, l in enumerate(label_defs)}
self.num_train = 0
self.num_valid = 0
self.num_test = 0
self.train_samples = []
self.valid_samples = []
import math
import cv2
import os
import numpy as np
from utils import Label, Box, Sample, Size
from utils import rgb2bgr, abs2prop
from glob import glob
from tqdm import tqdm
#-------------------------------------------------------------------------------
# Labels
#-------------------------------------------------------------------------------
label_defs = [
Label('aeroplane', rgb2bgr((0, 0, 0))),
Label('bicycle', rgb2bgr((111, 74, 0))),
Label('bird', rgb2bgr((81, 0, 81))),
Label('boat', rgb2bgr((128, 64, 128))),
Label('bottle', rgb2bgr((244, 35, 232))),
Label('bus', rgb2bgr((230, 150, 140))),
Label('car', rgb2bgr((70, 70, 70))),
Label('cat', rgb2bgr((102, 102, 156))),
Label('chair', rgb2bgr((190, 153, 153))),
Label('cow', rgb2bgr((150, 120, 90))),
Label('diningtable', rgb2bgr((153, 153, 153))),
Label('dog', rgb2bgr((250, 170, 30))),
Label('horse', rgb2bgr((220, 220, 0))),
Label('motorbike', rgb2bgr((107, 142, 35))),
Label('person', rgb2bgr((52, 151, 52))),
Label('pottedplant', rgb2bgr((70, 130, 180))),
import os
import imageio
import numpy as np
from utils import Label, Box, Sample, Size
from utils import rgb2bgr, abs2prop
from glob import glob
from tqdm import tqdm
#-------------------------------------------------------------------------------
# Labels
#-------------------------------------------------------------------------------
label_defs = [
#Label('aeroplane', rgb2bgr((1, 0, 0))),
Label('tumor', rgb2bgr((0, 0, 0))),
# Label('bicycle', rgb2bgr((111, 74, 0))),
# Label('bird', rgb2bgr(( 81, 0, 81))),
# Label('boat', rgb2bgr((128, 64, 128))),
# Label('bottle', rgb2bgr((244, 35, 232))),
# Label('bus', rgb2bgr((230, 150, 140))),
# Label('car', rgb2bgr(( 70, 70, 70))),
# Label('cat', rgb2bgr((102, 102, 156))),
# Label('chair', rgb2bgr((190, 153, 153))),
# Label('cow', rgb2bgr((150, 120, 90))),
# Label('diningtable', rgb2bgr((153, 153, 153))),
# Label('dog', rgb2bgr((250, 170, 30))),
# Label('horse', rgb2bgr((220, 220, 0))),
# Label('motorbike', rgb2bgr((107, 142, 35))),
# Label('person', rgb2bgr(( 52, 151, 52))),
# Label('pottedplant', rgb2bgr(( 70, 130, 180))),
from glob import glob
from tqdm import tqdm
#-------------------------------------------------------------------------------
# Labels
#-------------------------------------------------------------------------------
actions = [
'Basketball', 'BasketballDunk', 'Biking', 'CliffDiving', 'CricketBowling',
'Diving', 'Fencing', 'FloorGymnastics', 'GolfSwing', 'HorseRiding',
'IceDancing', 'LongJump', 'PoleVault', 'RopeClimbing', 'SalsaSpin',
'SkateBoarding', 'Skiing', 'Skijet', 'SoccerJuggling', 'Surfing',
'TennisSwing', 'TrampolineJumping', 'VolleyballSpiking', 'WalkingWithDog'
]
label_defs = [
Label('Basketball', rgb2bgr((0, 0, 0))),
Label('BasketballDunk', rgb2bgr((111, 74, 0))),
Label('Biking', rgb2bgr((81, 0, 81))),
Label('CliffDiving', rgb2bgr((128, 64, 128))),
Label('CricketBowling', rgb2bgr((244, 35, 232))),
Label('Diving', rgb2bgr((230, 150, 140))),
Label('Fencing', rgb2bgr((70, 70, 70))),
Label('FloorGymnastics', rgb2bgr((102, 102, 156))),
Label('GolfSwing', rgb2bgr((190, 153, 153))),
Label('HorseRiding', rgb2bgr((150, 120, 90))),
Label('IceDancing', rgb2bgr((153, 153, 153))),
Label('LongJump', rgb2bgr((250, 170, 30))),
Label('PoleVault', rgb2bgr((220, 220, 0))),
Label('RopeClimbing', rgb2bgr((107, 142, 35))),
Label('SalsaSpin', rgb2bgr((52, 151, 52))),
Label('SkateBoarding', rgb2bgr((70, 130, 180))),
import math
import cv2
import os
import numpy as np
from utils import Label, Box, Sample, Size
from utils import rgb2bgr, abs2prop
from glob import glob
from tqdm import tqdm
#-------------------------------------------------------------------------------
# Labels
#-------------------------------------------------------------------------------
label_defs = [
Label('hand', rgb2bgr((0, 0, 0)))
]
#-------------------------------------------------------------------------------
class PascalVOCSource:
#---------------------------------------------------------------------------
def __init__(self):
self.num_classes = len(label_defs)
self.colors = {l.name: l.color for l in label_defs}
self.lid2name = {i: l.name for i, l in enumerate(label_defs)}
self.lname2id = {l.name: i for i, l in enumerate(label_defs)}
self.num_train = 0
self.num_valid = 0
self.num_test = 0