Search
Tuesday, August 11, 2020
Here i would like to discuss the Covid-19 case prediction using tensorflow with transfer learning.
The dataset which am going to use which is download from the below link.
https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia
First import libraries..
import os
from tensorflow import keras
from tensorflow.keras import optimizers
from tensorflow.keras.models import Model, load_model
from tensorflow.keras.applications.mobilenet_v2 import MobileNetV2
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.layers import Add, Dense, GlobalAveragePooling2D, MaxPooling2D, Dropout
from tensorflow.keras.callbacks import ReduceLROnPlateau, EarlyStopping, ModelCheckpoint, LearningRateScheduler
import numpy as np
import matplotlib.pyplot as plt
Next load dataset and do the image augmentation process
# Specify training data directory
four_class_training_dir = 'all/train'