Skip to content

YOLOv5 object detection applied to finding corn crops and assessing planting quality by geometric means

License

Notifications You must be signed in to change notification settings

andre-kotze/kornrows

 
 

Repository files navigation

Yolov5 Custom Implementation for Crop Detection


YOLOv5 is a family of object detection architectures and models pretrained on the COCO dataset, and represents Ultralytics open-source research into future vision AI methods, incorporating lessons learned and best practices evolved over thousands of hours of research and development.
This is a customised version for corn plant identification as part of an ongoing research project.

Training Checkpoints

cornmodem864i12002
Colab trained model with 1,200 images (First50 dataset) containing 2,727 class instances
--img 864 --batch 8 --epochs 150 --cfg models/yolov5l.yaml
Peak performance in 23rd epoch
[email protected]:

kornmodel960m
Locally trained model with 684 images
--img 960 batch 4 --epochs 100 --data korn.yaml --cfg models/yolov5m.yaml
Computing time: ~31 hours
[email protected]: unknown

kornmodel960l_1200
Colab trained model with 1200 images
--img 960 --batch 8 --epochs 150 --data korn.yaml --cfg models/yolov5l.yaml
Computing time: ?
[email protected]: unknown


The Data

Labelled with labelImg


Workflow

  • Feed image into Yolov5 object detector trained on corn plant centres
  • Find the orientation of rows (well-differentiated lines in point cloud)
  • Find all rows with >= 2 plants
  • Measure variability in planting distance
  • Output Coefficient of Variability per row and per image

  • Utilities

      slice_images.py for resizing HR input images
      split_train_val.py for dividing data and structuring into Yolo-readable format
      count_instances.py to count class instances in labels
      find_kornrows.py to find the orientation of cornrows in the field

    About

    YOLOv5 object detection applied to finding corn crops and assessing planting quality by geometric means

    Topics

    Resources

    License

    Stars

    Watchers

    Forks

    Packages

    No packages published

    Languages

    • Python 98.3%
    • Other 1.7%