Delete model.py

model.py

@@ -1,44 +0,0 @@
-import os
-import numpy as np
-import tensorflow as tf
-from tensorflow.keras.layers import *
-
-#os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
-
-
-
-class OrientationClassifier(tf.keras.Model):
-    
-
-    def __init__(self,input_shape,hidden_dim,no_classes=1,isTrainable = False):
-        super().__init__()
-        resnet = tf.keras.applications.resnet50.ResNet50(include_top=False,input_shape=input_shape)
-        resnet.trainable = isTrainable
-
-        self.featureExtractor = tf.keras.models.Sequential([
-            resnet,
-            GlobalAveragePooling2D()
-        ])
-
-        self.classifier = tf.keras.models.Sequential([
-            Dense(hidden_dim,activation = 'relu'),
-            BatchNormalization(),
-            
-            Dense(hidden_dim/2,activation = 'relu'),
-            BatchNormalization(),
-            
-            Dense(hidden_dim/4,activation = 'relu'),
-            BatchNormalization(),
-            
-            Dense(no_classes,activation = 'sigmoid')
-        ])
-
-
-    def call(self, inputs):
-        
-        x = self.featureExtractor(inputs)
-        
-        preds = self.classifier(x)
-        
-        return preds
-