# 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