ZINDI - CGIAR Wheat Growth Stage Challenge by CGIAR Platform for Big Data in Agriculture

 FAST AI MODEL

#FastAI

import wandb

from wandb.fastai import WandbCallback  //// WEIGHT AND BIAS TO SEE INSIDE THE MODEL

wandb.init(project="wheat-zindi")

#HERE IMAGE NAMES IN CSV FILE AND IMAGES NAME AS COLUMN

import pandas as pd

from fastai.vision import *

train = pd.read_csv("./train.csv")

train['UID']  = train['UID'] + ".jpeg"

data  = ImageDataBunch.from_df("./Images",df=train, ds_tfms=get_transforms(),bs = 64,size = 224)

learn = cnn_learner(data, models.densenet121, metrics=[accuracy, top_k_accuracy], callback_fns=[WandbCallback])

learn.fit_one_cycle(4)