Update types.ts

types.ts

@@ -8,6 +8,8 @@ export enum LayerType {
   CONV3D = 'Conv3D',
   CONV_TRANSPOSE2D = 'ConvTranspose2D',
   DEFORMABLE_CONV = 'DeformConv2d',
+  SEPARABLE_CONV2D = 'SeparableConv2D', // New
+  DEPTHWISE_CONV2D = 'DepthwiseConv2D', // New
   RELU = 'ReLU',
   LEAKYRELU = 'LeakyReLU',
   PRELU = 'PReLU',
@@ -16,6 +18,8 @@ export enum LayerType {
   SWIGLU = 'SwiGLU',
   SIGMOID = 'Sigmoid',
   TANH = 'Tanh',
+  SOFTPLUS = 'Softplus', // New
+  SOFTSIGN = 'Softsign', // New
   MAXPOOL = 'MaxPool2D',
   MAXPOOL3D = 'MaxPool3D',
   AVGPOOL = 'AvgPool2D',
@@ -25,6 +29,7 @@ export enum LayerType {
   PIXEL_SHUFFLE = 'PixelShuffle',
   DROPOUT = 'Dropout',
   DROPPATH = 'DropPath',
+  SPATIAL_DROPOUT = 'SpatialDropout', // New
   BATCHNORM = 'BatchNorm2D',
   GROUPNORM = 'GroupNorm',
   LAYERNORM = 'LayerNorm',
@@ -53,6 +58,19 @@ export enum LayerType {
   SAM_PROMPT_ENCODER = 'SamPromptEncoder',
   SAM_MASK_DECODER = 'SamMaskDecoder',
   
+  // TensorFlow / Keras Preprocessing
+  RESCALING = 'Rescaling',
+  RESIZING = 'Resizing',
+  CENTER_CROP = 'CenterCrop',
+  RANDOM_FLIP = 'RandomFlip',
+  RANDOM_ROTATION = 'RandomRotation',
+  RANDOM_ZOOM = 'RandomZoom',
+  RANDOM_CONTRAST = 'RandomContrast',
+  TEXT_VECTORIZATION = 'TextVectorization',
+  NORMALIZATION_LAYER = 'NormalizationLayer', // Keras style mean/var
+  DISCRETIZATION = 'Discretization',
+  CATEGORY_ENCODING = 'CategoryEncoding',
+  
   // YOLO / Detection
   C2F_BLOCK = 'C2fBlock',
   SPPF_BLOCK = 'SPPFBlock',
@@ -138,8 +156,16 @@ export enum LayerType {
 
   CONCAT = 'Concat',
   ADD = 'Add',
+  SUBTRACT = 'Subtract', // New
+  MULTIPLY = 'Multiply', // New
+  AVERAGE = 'Average', // New
+  MAXIMUM = 'Maximum', // New
+  MINIMUM = 'Minimum', // New
+  DOT = 'Dot', // New
+  
   OUTPUT = 'Output',
   CUSTOM = 'CustomLayer',
+  LAMBDA = 'Lambda', // New
   IDENTITY = 'Identity'
 }
 
@@ -156,7 +182,7 @@ export interface LayerDefinition {
   type: LayerType;
   label: string;
   description: string;
-  category: 'Core' | 'Convolution' | 'Recurrent' | 'Normalization' | 'Transformer' | 'GenAI' | 'Utility' | 'Merge' | 'Detection' | 'Audio' | '3D' | 'OCR' | 'Robotics' | 'Video' | 'Graph' | 'Physics' | 'Spiking' | 'RL' | 'Advanced';
+  category: 'Core' | 'Convolution' | 'Preprocessing' | 'Recurrent' | 'Normalization' | 'Transformer' | 'GenAI' | 'Utility' | 'Merge' | 'Detection' | 'Audio' | '3D' | 'OCR' | 'Robotics' | 'Video' | 'Graph' | 'Physics' | 'Spiking' | 'RL' | 'Advanced';
   parameters: LayerParameter[];
 }