Release smoothing

src/captcha_detection/captcha_network.py

@@ -3,19 +3,9 @@
 import numpy as np
 import tensorflow as tf
 from tensorflow_core.python.keras.utils.vis_utils import plot_model
-from tensorflow.keras import backend as K
 
 from accuracy.correctly_classified_captcha_accuracy import CorrectlyClassifiedCaptchaAccuracy
-from dataset.captcha_dataset import CaptchaDataset
 from dataset.data_batcher import DataBatcher
-from dataset.preprocessing.image_preprocessors.image_cut_preprocessor import ImageCutPreprocessor
-from dataset.preprocessing.image_preprocessors.image_preprocessor_pipeline import ImagePreprocessorPipeline
-from dataset.preprocessing.image_preprocessors.normalize_image_preprocessor import NormalizeImagePreprocessor
-from dataset.preprocessing.labels_preprocessors.label_preprocess_pipeline import LabelPreprocessPipeline
-from dataset.preprocessing.labels_preprocessors.one_char_encoder import OneCharEncoder
-from dataset.preprocessing.labels_preprocessors.string_encoder import StringEncoder
-
-import sklearn.model_selection
 
 
 class CaptchaNetwork:

src/captcha_detection/test.py → src/test.py

@@ -3,7 +3,6 @@
 
 from dataset.preprocessing.image_preprocessors.convert_to_grayscale_preprocessor import ConvertToGrayscalePreprocessor
 from dataset.preprocessing.image_preprocessors.resize_preprocessor import ResizePreprocessor
-from utils.file_writer import FileWriter
 
 from dataset.preprocessing.image_preprocessors.image_preprocessor_pipeline import ImagePreprocessorPipeline
 from dataset.preprocessing.image_preprocessors.normalize_image_preprocessor import NormalizeImagePreprocessor
@@ -26,7 +25,7 @@
 if __name__ == "__main__":
     # Parse arguments
     parser = argparse.ArgumentParser()
-    parser.add_argument("--weights_file", default="src/captcha_detection/model.h5", type=str, help="Path to file that contains pre-trained weights.")
+    parser.add_argument("--weights_file", default="model.h5", type=str, help="Path to file that contains pre-trained weights.")
     parser.add_argument("--batch_size", default=32, type=int, help="Batch size.")
     parser.add_argument("--out_dir", default="out", type=str, help="Out dir")
     parser.add_argument("--captcha_length", default=4, type=int)

test.sh

@@ -1,3 +1,2 @@
 #!/bin/bash
-# this script can be run on cluster using command e.g.: qsub -l mem_free=8G,act_mem_free=8G,h_vmem=12G -cwd -j y sh train.sh
-/usr/bin/python3 ./src/captcha_detection/test.py
+venv/bin/python3 ./src/test.py --weights_file="src/captcha_detection/model.h5"

train.sh

@@ -1,6 +1,3 @@
 #!/bin/bash
-# this script can be run on cluster using command e.g.: qsub -l mem_free=8G,act_mem_free=8G,h_vmem=12G -cwd -j y sh train.sh
-/usr/bin/python3 ./src/captcha_detection/train.py \
---checkpoint_freq=1 \
---transformed_img_height=60 \
---transformed_img_width=60
+venv/bin/python3 src/train.py --weights_file="src/captcha_detection/model.h5" \
+--checkpoint_freq=1