Use yolov5 to detect apple conditions and provide apple rating.
The Apple condition detector detects the pests, cuts, bruises, and colors of apples and checks their grades in real time. You can use it if you want to know the grade of an apple or what pests you harvested are.
- Google colaboratory, local(GPU : 2070super 8GB, RAM : 16GB, CPU : 3700x)
- Yolov5
- Tensorflow 1.15.2v
- pytorch 1.10.0+cu111
- Android Studio @3.5.3
⚡️ Real-time Detection
Just take a video of an apple and it shows the conditions of the apple in real time.
🐜 Small object detection
During the image preprocessing step, even small objects can be detected by increasing the size of the image to 640
🍎 Detecting various types of apple pests
It can detect five pests that occur frequently in Fuji(부사사과).
🌈 Can Differentiate the colors of apples.
The apple condition detector can detect the color of the apple by dividing it into 3 colors.
🩹Detects many factors that affect apples
Various types of cuts and bruises that affect apple grade can be detected.
Data and number of annotations used in the apple condition detector :
- Image : 1533(After Data Augmentation)
- Class : 11
- ImageSize : 640
- Annotations : 7188
Images of apples were crawled by Google and Coupang, and pests used images provided by 🇰🇷NCPMS.
You can download apple_grade dataset here: https://app.roboflow.com/ds/GN9aUMJdR1?key=gcxVDmN2be
If you want to use the dataset from colab,
curl -L "https://app.roboflow.com/ds/5FcXgt0BjG?key=ElfkiFKHaY" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip- apple : It is a class to check if it is an apple(Using all kinds of apples)
- apple_A : An apple with superior color(Using only Fuji apples)
- apple_B : An apple with medium color(Using only Fuji apples)
- apple_C : An apple with bed color(Using only Fuji apples)
- dmg_s : Apple with small cuts or weak bruises that can be sold as a prize(Using all kinds of apples)
- dms_l : Apple with strong cuts or strong bruises that cannot be sold as a prize(Using all kinds of apples)
- Scab
- Anthracnose
- Sooty blotch
- Fly speck
- White rot
Sangmyung University Graduation Capstone Project
- Advisor : Professor Seongjoo Lee
- Team Leader : Kyuback Kim (email aderess : opea5954@gmail.com)
- Team member : Dongeun Kang
- Team member : Woonam Kim
Demo Video :














