Installation
Python Installation
requirements
basic installation:
extra python libs for parallel training
Docker
Usage
0. Check the config file by Visualization
1. Training
2. Evaluation
3. Export to ONNX or TRT model
Config System
MODEL
TRAIN
TRAIN.OPTIMIZER
TRAIN.LR_SCHEDULER
TEST
POST_PROCESS
DATASET
DATASET.PREPROC
Others
ssds.pytorch API
ssds.core
ssds.core.checkpoint
ssds.core.config
ssds.core.criterion
ssds.core.data_parallel
ssds.core.evaluation_metrics
ssds.core.optimizer
ssds.core.visualize_funcs
ssds.dataset
ssds.dataset.dataset_factory
ssds.dataset.detection_dataset
ssds.dataset.coco
ssds.dataset.dali_dataiterator
ssds.dataset.dali_coco
ssds.dataset.dali_tfrecord
ssds.dataset.transforms
ssds.modeling
ssds.modeling.layers
ssds.modeling.layers.box
ssds.modeling.layers.decoder
ssds.modeling.nets
ssds.modeling.nets.darknet
ssds.modeling.nets.densenet
ssds.modeling.nets.efficientnet
ssds.modeling.nets.effnet
ssds.modeling.nets.inception_v2
ssds.modeling.nets.mobilenet
ssds.modeling.nets.regnet
ssds.modeling.nets.resnet
ssds.modeling.nets.shufflenet
ssds.modeling.ssds
ssds.modeling.ssds.ssdsbase
ssds.modeling.ssds.ssd
ssds.modeling.ssds.yolo
ssds.modeling.ssds.fpn
ssds.modeling.ssds.bifpn
ssds.modeling.model_builder
ssds.pipeline
ssds.pipeline.pipeline_anchor_apex
ssds.pipeline.pipeline_anchor_basic
ssds.ssds
ssds.pytorch
»
Index
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
I
|
J
|
L
|
M
|
N
|
P
|
R
|
S
|
T
|
W
|
Y
_
__call__() (ssds.core.evaluation_metrics.MeanAveragePrecision method)
(ssds.modeling.layers.decoder.Decoder method)
(ssds.ssds.SSDDetector method)
__getitem__() (ssds.dataset.detection_dataset.DetectionDataset method)
A
add_anchorStrategy() (in module ssds.core.visualize_funcs)
add_defaultAnchors() (in module ssds.core.visualize_funcs)
add_extras() (ssds.modeling.ssds.SSD static method)
(ssds.modeling.ssds.SSDBiFPN static method)
(ssds.modeling.ssds.SSDFPN static method)
(ssds.modeling.ssds.YOLOV3 static method)
(ssds.modeling.ssds.YOLOV4 static method)
add_imagesWithBoxes() (in module ssds.core.visualize_funcs)
add_imagesWithMatchedBoxes() (in module ssds.core.visualize_funcs)
add_matchedAnchor() (in module ssds.core.visualize_funcs)
add_matchedAnchorsWithBox() (in module ssds.core.visualize_funcs)
add_prCurve() (in module ssds.core.visualize_funcs)
annotation_from_index() (ssds.dataset.coco.COCODataset method)
B
BalancedDataParallel (class in ssds.core.data_parallel)
box2delta() (in module ssds.modeling.layers.box)
C
cfg_from_file() (in module ssds.core.config)
CIOULoss() (in module ssds.core.criterion)
COCODataset (class in ssds.dataset.coco)
COCODetection (class in ssds.dataset.coco)
COCOPipeline (class in ssds.dataset.dali_coco)
Compose (class in ssds.dataset.transforms)
configure_lr_scheduler() (in module ssds.core.optimizer)
configure_optimizer() (in module ssds.core.optimizer)
configure_ratio_scale() (in module ssds.modeling.layers.box)
ConvertColor (class in ssds.dataset.transforms)
ConvertFromInts (class in ssds.dataset.transforms)
create_anchors() (in module ssds.modeling.model_builder)
create_decoder() (in module ssds.modeling.model_builder)
create_model() (in module ssds.modeling.model_builder)
D
DaliCOCO (class in ssds.dataset.dali_coco)
DaliDataset (class in ssds.dataset.dali_dataiterator)
DaliPipeline (class in ssds.dataset.dali_dataiterator)
DaliTFRecord (class in ssds.dataset.dali_tfrecord)
DarkNet53() (in module ssds.modeling.nets.darknet)
decode() (in module ssds.modeling.layers.box)
Decoder (class in ssds.modeling.layers.decoder)
define_graph() (ssds.dataset.dali_coco.COCOPipeline method)
(ssds.dataset.dali_tfrecord.TFRecordPipeline method)
delta2box() (in module ssds.modeling.layers.box)
delta2ltrb() (ssds.core.criterion.IOULoss method)
DenseNet121() (in module ssds.modeling.nets.densenet)
detection_collate() (in module ssds.dataset.dataset_factory)
DetectionDataset (class in ssds.dataset.detection_dataset)
DIOULoss() (in module ssds.core.criterion)
E
EfficientNetB0() (in module ssds.modeling.nets.efficientnet)
EfficientNetB1() (in module ssds.modeling.nets.efficientnet)
EfficientNetB2() (in module ssds.modeling.nets.efficientnet)
EfficientNetB3() (in module ssds.modeling.nets.efficientnet)
EfficientNetB4() (in module ssds.modeling.nets.efficientnet)
EfficientNetB5() (in module ssds.modeling.nets.efficientnet)
EffNetB0() (in module ssds.modeling.nets.effnet)
EffNetB0Ex() (in module ssds.modeling.nets.effnet)
EffNetB1() (in module ssds.modeling.nets.effnet)
EffNetB2() (in module ssds.modeling.nets.effnet)
EffNetB3() (in module ssds.modeling.nets.effnet)
EffNetB4() (in module ssds.modeling.nets.effnet)
EffNetB5() (in module ssds.modeling.nets.effnet)
EffNetB6() (in module ssds.modeling.nets.effnet)
EffNetB7() (in module ssds.modeling.nets.effnet)
eval_anchor_based_epoch() (in module ssds.pipeline.pipeline_anchor_basic)
Expand (class in ssds.dataset.transforms)
extract_targets() (in module ssds.modeling.layers.box)
F
find_previous_checkpoint() (in module ssds.core.checkpoint)
FocalLoss (class in ssds.core.criterion)
forward() (ssds.core.criterion.FocalLoss method)
(ssds.core.criterion.IOULoss method)
(ssds.core.criterion.MultiBoxLoss method)
(ssds.core.criterion.SmoothL1Loss method)
(ssds.modeling.ssds.SSD method)
(ssds.modeling.ssds.SSDBiFPN method)
(ssds.modeling.ssds.SSDFPN method)
(ssds.modeling.ssds.YOLOV3 method)
(ssds.modeling.ssds.YOLOV4 method)
G
generate_anchors() (in module ssds.modeling.layers.box)
get_results() (ssds.core.evaluation_metrics.MeanAveragePrecision method)
get_sample_region() (in module ssds.modeling.layers.box)
GIOULoss() (in module ssds.core.criterion)
I
image_path_from_index() (ssds.dataset.coco.COCODataset method)
InceptionV2 (class in ssds.modeling.nets.inception_v2)
initialize_extra() (ssds.modeling.ssds.ssdsbase.SSDSBase method)
initialize_head() (ssds.modeling.ssds.ssdsbase.SSDSBase method)
initialize_prior() (ssds.modeling.ssds.ssdsbase.SSDSBase method)
intersect() (in module ssds.dataset.transforms)
IOULoss (class in ssds.core.criterion)
J
jaccard_numpy() (in module ssds.dataset.transforms)
L
Lambda (class in ssds.dataset.transforms)
load_data() (in module ssds.dataset.dataset_factory)
M
MeanAveragePrecision (class in ssds.core.evaluation_metrics)
MobileNetV1() (in module ssds.modeling.nets.mobilenet)
MobileNetV2() (in module ssds.modeling.nets.mobilenet)
ModelWithLossBasic (class in ssds.pipeline.pipeline_anchor_apex)
module
ssds.core.checkpoint
ssds.core.config
ssds.core.criterion
ssds.core.optimizer
ssds.core.visualize_funcs
ssds.dataset.coco
ssds.dataset.dali_coco
ssds.dataset.dali_dataiterator
ssds.dataset.dali_tfrecord
ssds.dataset.dataset_factory
ssds.dataset.transforms
ssds.modeling.layers.box
ssds.modeling.model_builder
ssds.modeling.nets.darknet
ssds.modeling.nets.densenet
ssds.modeling.nets.efficientnet
ssds.modeling.nets.effnet
ssds.modeling.nets.mobilenet
ssds.modeling.nets.regnet
ssds.modeling.nets.resnet
ssds.modeling.nets.shufflenet
ssds.pipeline.pipeline_anchor_apex
ssds.pipeline.pipeline_anchor_basic
MultiBoxLoss (class in ssds.core.criterion)
N
nms() (in module ssds.modeling.layers.box)
Normalize (class in ssds.dataset.transforms)
P
PhotometricDistort (class in ssds.dataset.transforms)
predefined_graph() (ssds.dataset.dali_dataiterator.DaliPipeline method)
R
RandomBrightness (class in ssds.dataset.transforms)
RandomContrast (class in ssds.dataset.transforms)
RandomHue (class in ssds.dataset.transforms)
RandomLightingNoise (class in ssds.dataset.transforms)
RandomMirror (class in ssds.dataset.transforms)
RandomSampleCrop (class in ssds.dataset.transforms)
RandomSaturation (class in ssds.dataset.transforms)
RegNetX002() (in module ssds.modeling.nets.regnet)
RegNetX004() (in module ssds.modeling.nets.regnet)
RegNetX006() (in module ssds.modeling.nets.regnet)
RegNetX008() (in module ssds.modeling.nets.regnet)
RegNetX016() (in module ssds.modeling.nets.regnet)
RegNetX032() (in module ssds.modeling.nets.regnet)
RegNetX040() (in module ssds.modeling.nets.regnet)
RegNetX064() (in module ssds.modeling.nets.regnet)
RegNetX080() (in module ssds.modeling.nets.regnet)
RegNetX120() (in module ssds.modeling.nets.regnet)
RegNetX160() (in module ssds.modeling.nets.regnet)
RegNetX320() (in module ssds.modeling.nets.regnet)
Resize (class in ssds.dataset.transforms)
ResNet101() (in module ssds.modeling.nets.resnet)
ResNet152() (in module ssds.modeling.nets.resnet)
ResNet18() (in module ssds.modeling.nets.resnet)
ResNet34() (in module ssds.modeling.nets.resnet)
ResNet50() (in module ssds.modeling.nets.resnet)
ResNeXt101_32x8d() (in module ssds.modeling.nets.resnet)
ResNeXt50_32x4d() (in module ssds.modeling.nets.resnet)
resume_checkpoint() (in module ssds.core.checkpoint)
S
save_checkpoints() (in module ssds.core.checkpoint)
ShuffleNetV2_x1() (in module ssds.modeling.nets.shufflenet)
ShuffleNetV2_x2() (in module ssds.modeling.nets.shufflenet)
SmoothL1Loss (class in ssds.core.criterion)
snap_to_anchors_by_iou() (in module ssds.modeling.layers.box)
snap_to_anchors_by_scale() (in module ssds.modeling.layers.box)
SSD (class in ssds.modeling.ssds)
SSDBiFPN (class in ssds.modeling.ssds)
SSDDetector (class in ssds.ssds)
SSDFPN (class in ssds.modeling.ssds)
ssds.core.checkpoint
module
ssds.core.config
module
ssds.core.criterion
module
ssds.core.optimizer
module
ssds.core.visualize_funcs
module
ssds.dataset.coco
module
ssds.dataset.dali_coco
module
ssds.dataset.dali_dataiterator
module
ssds.dataset.dali_tfrecord
module
ssds.dataset.dataset_factory
module
ssds.dataset.transforms
module
ssds.modeling.layers.box
module
ssds.modeling.model_builder
module
ssds.modeling.nets.darknet
module
ssds.modeling.nets.densenet
module
ssds.modeling.nets.efficientnet
module
ssds.modeling.nets.effnet
module
ssds.modeling.nets.mobilenet
module
ssds.modeling.nets.regnet
module
ssds.modeling.nets.resnet
module
ssds.modeling.nets.shufflenet
module
ssds.pipeline.pipeline_anchor_apex
module
ssds.pipeline.pipeline_anchor_basic
module
SSDSBase (class in ssds.modeling.ssds.ssdsbase)
SwapChannels (class in ssds.dataset.transforms)
T
TFRecordPipeline (class in ssds.dataset.dali_tfrecord)
ToAbsoluteCoords (class in ssds.dataset.transforms)
ToCV2Image (class in ssds.dataset.transforms)
ToGPU (class in ssds.dataset.transforms)
ToPercentCoords (class in ssds.dataset.transforms)
ToTensor (class in ssds.dataset.transforms)
ToXYWH (class in ssds.dataset.transforms)
train_anchor_based_epoch() (in module ssds.pipeline.pipeline_anchor_apex)
(in module ssds.pipeline.pipeline_anchor_basic)
trainable_param() (in module ssds.core.optimizer)
W
WideResNet101_2() (in module ssds.modeling.nets.resnet)
WideResNet50_2() (in module ssds.modeling.nets.resnet)
Y
YOLOV3 (class in ssds.modeling.ssds)
YOLOV4 (class in ssds.modeling.ssds)