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

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git

Watch 4
Issues 62 Pull Requests 44 Discussions 0

#406 Feature/SG 293 - Add register loss

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-293-register_loss
Conversation 0 Files changed 4 Commits 1
File Compare
super-gradients
/
src
/
super_gradients
/
training
/
object_names.py
View left file View right file
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131 
  1. class Losses:
    2. 

  2.     """Static class holding all the supported loss names"""""
    3. 

  3.     CROSS_ENTROPY = "cross_entropy"
    4. 

  4.     MSE = "mse"
    5. 

  5.     R_SQUARED_LOSS = "r_squared_loss"
    6. 

  6.     SHELFNET_OHEM_LOSS = "shelfnet_ohem_loss"
    7. 

  7.     SHELFNET_SE_LOSS = "shelfnet_se_loss"
    8. 

  8.     YOLOX_LOSS = "yolox_loss"
    9. 

  9.     YOLOX_FAST_LOSS = "yolox_fast_loss"
    10. 

  10.     SSD_LOSS = "ssd_loss"
    11. 

  11.     STDC_LOSS = "stdc_loss"
    12. 

  12.     BCE_DICE_LOSS = "bce_dice_loss"
    13. 

  13.     KD_LOSS = "kd_loss"
    14. 

  14.     DICE_CE_EDGE_LOSS = "dice_ce_edge_loss"
    15. 

  15. 

  16. 

  17. class Metrics:
    18. 

  18.     """Static class holding all the supported metric names"""""
    19. 

  19.     ACCURACY = 'Accuracy'
    20. 

  20.     TOP5 = 'Top5'
    21. 

  21.     DETECTION_METRICS = 'DetectionMetrics'
    22. 

  22.     IOU = 'IoU'
    23. 

  23.     BINARY_IOU = "BinaryIOU"
    24. 

  24.     DICE = "Dice"
    25. 

  25.     BINARY_DICE = "BinaryDice"
    26. 

  26.     PIXEL_ACCURACY = 'PixelAccuracy'
    27. 

  27. 

  28. 

  29. class Transforms:
    30. 

  30.     """Static class holding all the supported transform names"""""
    31. 

  31.     # From SG
    32. 

  32.     SegRandomFlip = "SegRandomFlip"
    33. 

  33.     SegResize = "SegResize"
    34. 

  34.     SegRescale = "SegRescale"
    35. 

  35.     SegRandomRescale = "SegRandomRescale"
    36. 

  36.     SegRandomRotate = "SegRandomRotate"
    37. 

  37.     SegCropImageAndMask = "SegCropImageAndMask"
    38. 

  38.     SegRandomGaussianBlur = "SegRandomGaussianBlur"
    39. 

  39.     SegPadShortToCropSize = "SegPadShortToCropSize"
    40. 

  40.     SegColorJitter = "SegColorJitter"
    41. 

  41.     DetectionMosaic = "DetectionMosaic"
    42. 

  42.     DetectionRandomAffine = "DetectionRandomAffine"
    43. 

  43.     DetectionMixup = "DetectionMixup"
    44. 

  44.     DetectionHSV = "DetectionHSV"
    45. 

  45.     DetectionHorizontalFlip = "DetectionHorizontalFlip"
    46. 

  46.     DetectionPaddedRescale = "DetectionPaddedRescale"
    47. 

  47.     DetectionTargetsFormat = "DetectionTargetsFormat"
    48. 

  48.     DetectionTargetsFormatTransform = "DetectionTargetsFormatTransform"
    49. 

  49.     RandomResizedCropAndInterpolation = "RandomResizedCropAndInterpolation"
    50. 

  50.     RandAugmentTransform = "RandAugmentTransform"
    51. 

  51.     Lighting = "Lighting"
    52. 

  52.     RandomErase = "RandomErase"
    53. 

  53. 

  54.     # From torch
    55. 

  55.     Compose = "Compose"
    56. 

  56.     ToTensor = "ToTensor"
    57. 

  57.     PILToTensor = "PILToTensor"
    58. 

  58.     ConvertImageDtype = "ConvertImageDtype"
    59. 

  59.     ToPILImage = "ToPILImage"
    60. 

  60.     Normalize = "Normalize"
    61. 

  61.     Resize = "Resize"
    62. 

  62.     CenterCrop = "CenterCrop"
    63. 

  63.     Pad = "Pad"
    64. 

  64.     Lambda = "Lambda"
    65. 

  65.     RandomApply = "RandomApply"
    66. 

  66.     RandomChoice = "RandomChoice"
    67. 

  67.     RandomOrder = "RandomOrder"
    68. 

  68.     RandomCrop = "RandomCrop"
    69. 

  69.     RandomHorizontalFlip = "RandomHorizontalFlip"
    70. 

  70.     RandomVerticalFlip = "RandomVerticalFlip"
    71. 

  71.     RandomResizedCrop = "RandomResizedCrop"
    72. 

  72.     FiveCrop = "FiveCrop"
    73. 

  73.     TenCrop = "TenCrop"
    74. 

  74.     LinearTransformation = "LinearTransformation"
    75. 

  75.     ColorJitter = "ColorJitter"
    76. 

  76.     RandomRotation = "RandomRotation"
    77. 

  77.     RandomAffine = "RandomAffine"
    78. 

  78.     Grayscale = "Grayscale"
    79. 

  79.     RandomGrayscale = "RandomGrayscale"
    80. 

  80.     RandomPerspective = "RandomPerspective"
    81. 

  81.     RandomErasing = "RandomErasing"
    82. 

  82.     GaussianBlur = "GaussianBlur"
    83. 

  83.     InterpolationMode = "InterpolationMode"
    84. 

  84.     RandomInvert = "RandomInvert"
    85. 

  85.     RandomPosterize = "RandomPosterize"
    86. 

  86.     RandomSolarize = "RandomSolarize"
    87. 

  87.     RandomAdjustSharpness = "RandomAdjustSharpness"
    88. 

  88.     RandomAutocontrast = "RandomAutocontrast"
    89. 

  89.     RandomEqualize = "RandomEqualize"
    90. 

  90. 

  91. 

  92. class Optimizers:
    93. 

  93.     """Static class holding all the supported optimizer names"""""
    94. 

  94.     SGD = "SGD"
    95. 

  95.     ADAM = "Adam"
    96. 

  96.     RMS_PROP = "RMSprop"
    97. 

  97.     RMS_PROP_TF = "RMSpropTF"
    98. 

  98.     LAMB = "Lamb"
    99. 

  99. 

  100. 

  101. class Callbacks:
    102. 

  102.     """Static class holding all the supported callback names"""""
    103. 

  103.     DECI_LAB_UPLOAD = 'DeciLabUploadCallback'
    104. 

  104.     LR_CALLBACK_BASE = 'LRCallbackBase'
    105. 

  105.     LR_SCHEDULER = 'LRSchedulerCallback'
    106. 

  106.     METRICS_UPDATE = 'MetricsUpdateCallback'
    107. 

  107.     MODEL_CONVERSION_CHECK = 'ModelConversionCheckCallback'
    108. 

  108.     EARLY_STOP = 'EarlyStop'
    109. 

  109.     DETECTION_MULTISCALE_PREPREDICTION = 'DetectionMultiscalePrePredictionCallback'
    110. 

  110.     YOLOX_TRAINING_STAGE_SWITCH = 'YoloXTrainingStageSwitchCallback'
    111. 

  111. 

  112. 

  113. class LRSchedulers:
    114. 

  114.     """Static class to hold all the supported LR Scheduler names"""""
    115. 

  115.     STEP = "step"
    116. 

  116.     POLY = "poly"
    117. 

  117.     COSINE = "cosine"
    118. 

  118.     EXP = "exp"
    119. 

  119.     FUNCTION = "function"
    120. 

  120. 

  121. 

  122. class LRWarmups:
    123. 

  123.     """Static class to hold all the supported LR Warmup names"""""
    124. 

  124.     LINEAR_STEP = "linear_step"
    125. 

  125. 

  126. 

  127. class Samplers:
    128. 

  128.     """Static class to hold all the supported Samplers names"""""
    129. 

  129.     INFINITE = "InfiniteSampler"
    130. 

  130.     REPEAT_AUG = "RepeatAugSampler"
    131. 

  131.     DISTRIBUTED = "DistributedSampler"
    132.
Discard
Prev
Next
Tip!

Press p or to see the previous file or, n or to see the next file