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 inherited from Detailed Description
Definition at line 45 of file train_mask_rcnn.py.
Constructor & Destructor Documentation
◆ __init__()
| def train_mask_rcnn.TrainMaskRCNN.__init__ | ( | self | ) |
Definition at line 47 of file train_mask_rcnn.py.
Member Function Documentation
◆ load_dataset()
| def train_mask_rcnn.TrainMaskRCNN.load_dataset | ( | self | ) |
Definition at line 135 of file train_mask_rcnn.py.
◆ load_model()
| def train_mask_rcnn.TrainMaskRCNN.load_model | ( | self | ) |
Definition at line 140 of file train_mask_rcnn.py.
◆ setup_iterator()
| def train_mask_rcnn.TrainMaskRCNN.setup_iterator | ( | self | ) |
Definition at line 199 of file train_mask_rcnn.py.
◆ setup_optimizer()
| def train_mask_rcnn.TrainMaskRCNN.setup_optimizer | ( | self | ) |
Definition at line 182 of file train_mask_rcnn.py.
◆ setup_trainer()
| def train_mask_rcnn.TrainMaskRCNN.setup_trainer | ( | self | ) |
Definition at line 212 of file train_mask_rcnn.py.
Member Data Documentation
◆ batch_size
| train_mask_rcnn.TrainMaskRCNN.batch_size |
Definition at line 103 of file train_mask_rcnn.py.
◆ eval_interval
| train_mask_rcnn.TrainMaskRCNN.eval_interval |
Definition at line 115 of file train_mask_rcnn.py.
◆ eval_interval_type
| train_mask_rcnn.TrainMaskRCNN.eval_interval_type |
Definition at line 116 of file train_mask_rcnn.py.
◆ gpu
| train_mask_rcnn.TrainMaskRCNN.gpu |
Definition at line 102 of file train_mask_rcnn.py.
◆ log_interval
| train_mask_rcnn.TrainMaskRCNN.log_interval |
Definition at line 111 of file train_mask_rcnn.py.
◆ log_interval_type
| train_mask_rcnn.TrainMaskRCNN.log_interval_type |
Definition at line 112 of file train_mask_rcnn.py.
◆ lr
| train_mask_rcnn.TrainMaskRCNN.lr |
Definition at line 105 of file train_mask_rcnn.py.
◆ mask_rcnn
| train_mask_rcnn.TrainMaskRCNN.mask_rcnn |
Definition at line 151 of file train_mask_rcnn.py.
◆ max_epoch
| train_mask_rcnn.TrainMaskRCNN.max_epoch |
Definition at line 104 of file train_mask_rcnn.py.
◆ model
| train_mask_rcnn.TrainMaskRCNN.model |
Definition at line 176 of file train_mask_rcnn.py.
◆ model_name
| train_mask_rcnn.TrainMaskRCNN.model_name |
Definition at line 101 of file train_mask_rcnn.py.
◆ optimizer
| train_mask_rcnn.TrainMaskRCNN.optimizer |
Definition at line 183 of file train_mask_rcnn.py.
◆ out_dir
| train_mask_rcnn.TrainMaskRCNN.out_dir |
Definition at line 107 of file train_mask_rcnn.py.
◆ plot_interval
| train_mask_rcnn.TrainMaskRCNN.plot_interval |
Definition at line 113 of file train_mask_rcnn.py.
◆ plot_interval_type
| train_mask_rcnn.TrainMaskRCNN.plot_interval_type |
Definition at line 114 of file train_mask_rcnn.py.
◆ print_interval
| train_mask_rcnn.TrainMaskRCNN.print_interval |
Definition at line 109 of file train_mask_rcnn.py.
◆ print_interval_type
| train_mask_rcnn.TrainMaskRCNN.print_interval_type |
Definition at line 110 of file train_mask_rcnn.py.
◆ progressbar_update_interval
| train_mask_rcnn.TrainMaskRCNN.progressbar_update_interval |
Definition at line 108 of file train_mask_rcnn.py.
◆ save_interval
| train_mask_rcnn.TrainMaskRCNN.save_interval |
Definition at line 117 of file train_mask_rcnn.py.
◆ save_interval_type
| train_mask_rcnn.TrainMaskRCNN.save_interval_type |
Definition at line 118 of file train_mask_rcnn.py.
◆ timestamp_iso
| train_mask_rcnn.TrainMaskRCNN.timestamp_iso |
Definition at line 121 of file train_mask_rcnn.py.
◆ train_dataset
| train_mask_rcnn.TrainMaskRCNN.train_dataset |
Definition at line 136 of file train_mask_rcnn.py.
◆ train_dataset_dir
| train_mask_rcnn.TrainMaskRCNN.train_dataset_dir |
Definition at line 99 of file train_mask_rcnn.py.
◆ train_iterator
| train_mask_rcnn.TrainMaskRCNN.train_iterator |
Definition at line 206 of file train_mask_rcnn.py.
◆ trainer
| train_mask_rcnn.TrainMaskRCNN.trainer |
Definition at line 223 of file train_mask_rcnn.py.
◆ updater
| train_mask_rcnn.TrainMaskRCNN.updater |
Definition at line 220 of file train_mask_rcnn.py.
◆ val_dataset
| train_mask_rcnn.TrainMaskRCNN.val_dataset |
Definition at line 138 of file train_mask_rcnn.py.
◆ val_dataset_dir
| train_mask_rcnn.TrainMaskRCNN.val_dataset_dir |
Definition at line 100 of file train_mask_rcnn.py.
◆ val_iterator
| train_mask_rcnn.TrainMaskRCNN.val_iterator |
Definition at line 208 of file train_mask_rcnn.py.
◆ weight_decay
| 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: