Global Wheat Detection with EfficientDet

About The Project

This is my solution to the Global Wheat Detection that uses EfficientDet.

Getting Started

To get a local copy up and running, follow these simple example steps.

Installation

1. Clone the repo

   git clone https://github.com/stephenllh/wheat_efficientdet.git

2. Change directory

   cd wheat_efficientdet

3. Install packages

   pip install requirements.txt

Usage

1. Create a directory called input

   mkdir input
   cd input

2. Download the dataset into the folder

   • Option 1: Use Kaggle API
     • pip install kaggle
     • kaggle competitions download -c global-wheat-detection
   • Option 2: Download the dataset from the competition website.

3. Run the training script

   cd ..
   python train.py

4. (Optional) Run the inference script

   python inference.py

License

Distributed under the MIT License. See LICENSE for more information.

Contact

Stephen Lau - Email - Twitter - Kaggle