def train_ANN( self): #training neural networks with examples from chars74k-lite self.classification = Classification.Classifier() self.classification.train_neural_network() _ = joblib.dump(self.classification.network, "classification_weights.pkl", compress=9)
def main(): classification = Classification.Classifier() #Classification part classification.do_knn() classification.do_svm() classification.do_random_forest() classification.do_neural_network() detection = Detection.Detector() #sliding window and classification detection.run()
def load_weights(self): #used for mobility in testing self.classification = Classification.Classifier() self.classification.network = joblib.load("classification_weights.pkl")