Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#643 PPYolo-E

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-344-PP-Yolo-E-Training-Replicate-Recipe
1 changed files with 9 additions and 0 deletions
  1. 9
    0
      src/super_gradients/common/object_names.py
@@ -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:
Discard