ssd_train_dataset Namespace Reference

Classes
class	MultiboxTrainChain
class	SSDDataset
class	Transform

Variables
	args = p.parse_args()
	dataset = SSDDataset(args.train, label_names)
	default
	file
	float
	help
	int
	label_names = tuple(yaml.load(f))
string	log_interval = args.log_iter, "iteration"
	model = SSD300(n_fg_class=len(dataset.label_names))
	optimizer = chainer.optimizers.MomentumSGD(lr=args.lr)
	p = argparse.ArgumentParser()
	pretrained_model = SSD300(pretrained_model=args.base_model)
	str
	test = SSDDataset(args.val, label_names)
	test_iter
	train = SSDDataset(args.train, label_names)
	train_chain = MultiboxTrainChain(model)
	train_iter
	trainer
	trigger
	type
	updater
string	val_interval = args.val_iter, "iteration"

Variable Documentation

args

ssd_train_dataset.args = p.parse_args()

Definition at line 185 of file ssd_train_dataset.py.

dataset

ssd_train_dataset.dataset = SSDDataset(args.train, label_names)

Definition at line 194 of file ssd_train_dataset.py.

default

ssd_train_dataset.default

Definition at line 173 of file ssd_train_dataset.py.

file

ssd_train_dataset.file

Definition at line 25 of file ssd_train_dataset.py.

float

ssd_train_dataset.float

Definition at line 180 of file ssd_train_dataset.py.

help

ssd_train_dataset.help

Definition at line 171 of file ssd_train_dataset.py.

int

ssd_train_dataset.int

Definition at line 175 of file ssd_train_dataset.py.

label_names

ssd_train_dataset.label_names = tuple(yaml.load(f))

Definition at line 189 of file ssd_train_dataset.py.

log_interval

string ssd_train_dataset.log_interval = args.log_iter, "iteration"

Definition at line 248 of file ssd_train_dataset.py.

model

ssd_train_dataset.model = SSD300(n_fg_class=len(dataset.label_names))

Definition at line 206 of file ssd_train_dataset.py.

optimizer

ssd_train_dataset.optimizer = chainer.optimizers.MomentumSGD(lr=args.lr)

Definition at line 227 of file ssd_train_dataset.py.

p

ssd_train_dataset.p = argparse.ArgumentParser()

Definition at line 170 of file ssd_train_dataset.py.

pretrained_model

ssd_train_dataset.pretrained_model = SSD300(pretrained_model=args.base_model)

Definition at line 203 of file ssd_train_dataset.py.

str

ssd_train_dataset.str

Definition at line 178 of file ssd_train_dataset.py.

test

ssd_train_dataset.test = SSDDataset(args.val, label_names)

Definition at line 195 of file ssd_train_dataset.py.

test_iter

ssd_train_dataset.test_iter

Initial value:

=  chainer.iterators.SerialIterator(
        test, args.batchsize,
        repeat=False, shuffle=False)

Definition at line 223 of file ssd_train_dataset.py.

train

ssd_train_dataset.train = SSDDataset(args.train, label_names)

Definition at line 195 of file ssd_train_dataset.py.

train_chain

ssd_train_dataset.train_chain = MultiboxTrainChain(model)

Definition at line 212 of file ssd_train_dataset.py.

train_iter

ssd_train_dataset.train_iter

Initial value:

=  chainer.iterators.MultiprocessIterator(
        train, args.batchsize)

Definition at line 220 of file ssd_train_dataset.py.

trainer

ssd_train_dataset.trainer

Initial value:

=  training.Trainer(
        updater, (args.iteration, "iteration"), args.out)

Definition at line 238 of file ssd_train_dataset.py.

trigger

ssd_train_dataset.trigger

Definition at line 246 of file ssd_train_dataset.py.

type

ssd_train_dataset.type

Definition at line 175 of file ssd_train_dataset.py.

updater

ssd_train_dataset.updater

Initial value:

=  training.StandardUpdater(
        train_iter, optimizer, device=args.gpu)

Definition at line 236 of file ssd_train_dataset.py.

val_interval

string ssd_train_dataset.val_interval = args.val_iter, "iteration"

Definition at line 241 of file ssd_train_dataset.py.