Classes
class	MultiboxTrainChain
class	SSDDataset
class	Transform
Variables
tuple	args = p.parse_args()
tuple	dataset = SSDDataset(args.train, label_names)
tuple	label_names = tuple(yaml.load(f))
string	log_interval = "iteration"
tuple	model = SSD300(n_fg_class=len(dataset.label_names))
tuple	optimizer = chainer.optimizers.MomentumSGD(lr=args.lr)
tuple	p = argparse.ArgumentParser()
tuple	pretrained_model = SSD300(pretrained_model=args.base_model)
tuple	test = SSDDataset(args.val, label_names)
tuple	test_iter
tuple	train = SSDDataset(args.train, label_names)
tuple	train_chain = MultiboxTrainChain(model)
tuple	train_iter
tuple	trainer
	trigger = val_interval)
tuple	updater
string	val_interval = "iteration"

Variable Documentation

Definition at line 163 of file ssd_train_dataset.py.

Definition at line 172 of file ssd_train_dataset.py.

Definition at line 167 of file ssd_train_dataset.py.

Definition at line 226 of file ssd_train_dataset.py.

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

Definition at line 184 of file ssd_train_dataset.py.

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

Definition at line 205 of file ssd_train_dataset.py.

tuple ssd_train_dataset::p = argparse.ArgumentParser()

Definition at line 148 of file ssd_train_dataset.py.

Definition at line 181 of file ssd_train_dataset.py.

Definition at line 177 of file ssd_train_dataset.py.

Initial value:

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

Definition at line 201 of file ssd_train_dataset.py.

Definition at line 176 of file ssd_train_dataset.py.

Definition at line 190 of file ssd_train_dataset.py.

Initial value:

00001 chainer.iterators.MultiprocessIterator(
00002         train, args.batchsize)

Definition at line 198 of file ssd_train_dataset.py.

Initial value:

00001 training.Trainer(
00002         updater, (args.iteration, "iteration"), args.out)

Definition at line 216 of file ssd_train_dataset.py.

Definition at line 224 of file ssd_train_dataset.py.

Initial value:

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

Definition at line 214 of file ssd_train_dataset.py.

Definition at line 219 of file ssd_train_dataset.py.

Classes