train_mask_rcnn.TrainMaskRCNN Class Reference

Inheritance diagram for train_mask_rcnn.TrainMaskRCNN:

Public Member Functions
def	__init__ (self)
def	load_dataset (self)
def	load_model (self)
def	setup_iterator (self)
def	setup_optimizer (self)
def	setup_trainer (self)

Public Attributes
	batch_size
	eval_interval
	eval_interval_type
	gpu
	log_interval
	log_interval_type
	lr
	mask_rcnn
	max_epoch
	model
	model_name
	optimizer
	out_dir
	plot_interval
	plot_interval_type
	print_interval
	print_interval_type
	progressbar_update_interval
	save_interval
	save_interval_type
	timestamp_iso
	train_dataset
	train_dataset_dir
	train_iterator
	trainer
	updater
	val_dataset
	val_dataset_dir
	val_iterator
	weight_decay
Public Attributes inherited from object
pointer	iv [2]

Detailed Description

Definition at line 45 of file train_mask_rcnn.py.

Constructor & Destructor Documentation

def train_mask_rcnn.TrainMaskRCNN.__init__

(

self

)

Definition at line 47 of file train_mask_rcnn.py.

Member Function Documentation

def train_mask_rcnn.TrainMaskRCNN.load_dataset

(

self

)

Definition at line 135 of file train_mask_rcnn.py.

def train_mask_rcnn.TrainMaskRCNN.load_model

(

self

)

Definition at line 140 of file train_mask_rcnn.py.

def train_mask_rcnn.TrainMaskRCNN.setup_iterator

(

self

)

Definition at line 199 of file train_mask_rcnn.py.

def train_mask_rcnn.TrainMaskRCNN.setup_optimizer

(

self

)

Definition at line 182 of file train_mask_rcnn.py.

def train_mask_rcnn.TrainMaskRCNN.setup_trainer

(

self

)

Definition at line 212 of file train_mask_rcnn.py.

Member Data Documentation

train_mask_rcnn.TrainMaskRCNN.batch_size

Definition at line 103 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.eval_interval

Definition at line 115 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.eval_interval_type

Definition at line 116 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.gpu

Definition at line 102 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.log_interval

Definition at line 111 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.log_interval_type

Definition at line 112 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.lr

Definition at line 105 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.mask_rcnn

Definition at line 151 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.max_epoch

Definition at line 104 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.model

Definition at line 176 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.model_name

Definition at line 101 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.optimizer

Definition at line 183 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.out_dir

Definition at line 107 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.plot_interval

Definition at line 113 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.plot_interval_type

Definition at line 114 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.print_interval

Definition at line 109 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.print_interval_type

Definition at line 110 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.progressbar_update_interval

Definition at line 108 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.save_interval

Definition at line 117 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.save_interval_type

Definition at line 118 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.timestamp_iso

Definition at line 121 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.train_dataset

Definition at line 136 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.train_dataset_dir

Definition at line 99 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.train_iterator

Definition at line 206 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.trainer

Definition at line 223 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.updater

Definition at line 220 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.val_dataset

Definition at line 138 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.val_dataset_dir

Definition at line 100 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.val_iterator

Definition at line 208 of file train_mask_rcnn.py.

train_mask_rcnn.TrainMaskRCNN.weight_decay

Definition at line 106 of file train_mask_rcnn.py.

---

The documentation for this class was generated from the following file:

• train_mask_rcnn.py