ssd_train_dataset.py File Reference

Go to the source code of this file.

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