|
@@ -7,6 +7,7 @@ class Losses:
|
|
SHELFNET_OHEM_LOSS = "shelfnet_ohem_loss"
|
|
SHELFNET_OHEM_LOSS = "shelfnet_ohem_loss"
|
|
SHELFNET_SE_LOSS = "shelfnet_se_loss"
|
|
SHELFNET_SE_LOSS = "shelfnet_se_loss"
|
|
YOLOX_LOSS = "yolox_loss"
|
|
YOLOX_LOSS = "yolox_loss"
|
|
|
|
+ PPYOLOE_LOSS = "ppyoloe_loss"
|
|
YOLOX_FAST_LOSS = "yolox_fast_loss"
|
|
YOLOX_FAST_LOSS = "yolox_fast_loss"
|
|
SSD_LOSS = "ssd_loss"
|
|
SSD_LOSS = "ssd_loss"
|
|
STDC_LOSS = "stdc_loss"
|
|
STDC_LOSS = "stdc_loss"
|
|
@@ -55,6 +56,8 @@ class Transforms:
|
|
DetectionRescale = "DetectionRescale"
|
|
DetectionRescale = "DetectionRescale"
|
|
DetectionPaddedRescale = "DetectionPaddedRescale"
|
|
DetectionPaddedRescale = "DetectionPaddedRescale"
|
|
DetectionTargetsFormatTransform = "DetectionTargetsFormatTransform"
|
|
DetectionTargetsFormatTransform = "DetectionTargetsFormatTransform"
|
|
|
|
+ DetectionNormalize = "DetectionNormalize"
|
|
|
|
+ #
|
|
RandomResizedCropAndInterpolation = "RandomResizedCropAndInterpolation"
|
|
RandomResizedCropAndInterpolation = "RandomResizedCropAndInterpolation"
|
|
RandAugmentTransform = "RandAugmentTransform"
|
|
RandAugmentTransform = "RandAugmentTransform"
|
|
Lighting = "Lighting"
|
|
Lighting = "Lighting"
|
|
@@ -131,6 +134,8 @@ class Callbacks:
|
|
EARLY_STOP = "EarlyStop"
|
|
EARLY_STOP = "EarlyStop"
|
|
DETECTION_MULTISCALE_PREPREDICTION = "DetectionMultiscalePrePredictionCallback"
|
|
DETECTION_MULTISCALE_PREPREDICTION = "DetectionMultiscalePrePredictionCallback"
|
|
YOLOX_TRAINING_STAGE_SWITCH = "YoloXTrainingStageSwitchCallback"
|
|
YOLOX_TRAINING_STAGE_SWITCH = "YoloXTrainingStageSwitchCallback"
|
|
|
|
+ PPYOLOE_TRAINING_STAGE_SWITCH = "PPYoloETrainingStageSwitchCallback"
|
|
|
|
+ DETECTION_VISUALIZATION_CALLBACK = "DetectionVisualizationCallback"
|
|
|
|
|
|
|
|
|
|
class LRSchedulers:
|
|
class LRSchedulers:
|
|
@@ -275,6 +280,10 @@ class Models:
|
|
UNET_CUSTOM_CLS = "unet_custom_cls"
|
|
UNET_CUSTOM_CLS = "unet_custom_cls"
|
|
STDC_CUSTOM = "stdc_custom"
|
|
STDC_CUSTOM = "stdc_custom"
|
|
STDC_CUSTOM_CLS = "stdc_custom_cls"
|
|
STDC_CUSTOM_CLS = "stdc_custom_cls"
|
|
|
|
+ PP_YOLOE_S = "ppyoloe_s"
|
|
|
|
+ PP_YOLOE_M = "ppyoloe_m"
|
|
|
|
+ PP_YOLOE_L = "ppyoloe_l"
|
|
|
|
+ PP_YOLOE_X = "ppyoloe_x"
|
|
|
|
|
|
|
|
|
|
class ConcatenatedTensorFormats:
|
|
class ConcatenatedTensorFormats:
|