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'







Leave a Reply

Subscribe to Posts | Subscribe to Comments