1

TRƯỜNG ĐẠI HỌC SƯ PHẠM KỸ THUẬT THÀNH PHỐ HỒ CHÍ MINH
KHOA ĐÀO TẠO CHẤT LƯỢNG CAO

ĐỒ ÁN TỐT NGHIỆP

THIẾT KẾ THI CÔNG MÔ HÌNH PHÂN LOẠI TRÁI
CÂY THEO MÀU SẮC

SVTH :
MSSV :
Khoá :
Ngành :
GVHD:

CAO HOÀNG ANH KHA
18142128
2018
CÔNG NGHỆ KỸ THUẬT ĐIỆN-ĐIỆN TỬ
TS. NGUYỄN THỊ LƯỠNG

Tp. Hồ Chí Minh, tháng 12 năm 2022

CỘNG HÒA XÃ HỘI CHỦ NGHĨA VIỆT NAM
Độc lập – Tự do – Hạnh phúc
***

PHIẾU NHẬN XÉT CỦA GIÁO VIÊN HƯỚNG DẪN
Họ và tên Sinh viên: Cao Hoàng Anh Kha

MSSV: 18142128

Ngành: Công nghệ Kỹ thuật Điện-Điện tử.
Tên đề tài:
THIẾT KẾ - THI CÔNG MÔ HÌNH PHÂN LOẠI TRÁI CÂY THEO MÀU SẮC.
Họ và tên Giáo viên hướng dẫn:
................................................................................................................................................
NHẬN XÉT:
1. Về nội dung đề tài &amp khối lượng thực hiện:
................................................................................................................................................
................................................................................................................................................
2. Ưu điểm:
...............................................................................................................................................
...............................................................................................................................................
................................................................................................................................................
3. Khuyết điểm:
................................................................................................................................................
................................................................................................................................................
...............................................................................................................................................
4. Đề nghị cho bảo vệ hay không?
................................................................................................................................................
5. Ý kiến khác:
...............................................................................................................................................
6. Điểm:……………….(Bằng chữ:....................................................................
...

...


63

Code Raspberry:
import cv2
import time
import serial
Conf_threshold = 0.7
NMS_threshold = 0.4
COLORS = [(0, 255, 0), (0, 0, 255), (255, 0, 0),
(255, 255, 0), (255, 0, 255), (0, 255, 255)]
class_name = []
with open(/home/pi/YOLOV4_fastapi/classes.txt, r) as f:
class_name = [cname.strip() for cname in f.readlines()]
#print(class_name)
class Detection():
def __init__(self):
self.net = cv2.dnn.readNetFromDarknet(/home/pi/YOLOV4_fastapi/yolov4tiny.cfg, /home/pi/YOLOV4_fastapi/yolov4-tiny_final.weights)
self.model = cv2.dnn_DetectionModel(self.net)
self.model.setInputParams(size=(416, 416), scale=1/255.0, swapRB=True)
self.count=0
self.count_taodo=0
self.count_taoxanh=0
self.count_chuoixanh=0
self.count_chuoichin=0
self.sr = serial.Serial(/dev/ttyACM0,9600)
self.sr.flushInput()
print(Starting Serial)
def detect(self,frame):

classes, scores, boxes = self.model.detect(frame, Conf_threshold,
NMS_threshold)
64

if len(boxes)0:
for (classid, score, box) in zip(classes, scores, boxes):
color = COLORS[int(classid) % len(COLORS)]
label = .format(class_name[int(classid)])
msg=class_name[classid[0]]+\\n
print(msg)
print(score)
cv2.rectangle(frame, box, color, 1)
cv2.putText(frame, label, (box[0], box[1]-10),
cv2.FONT_HERSHEY_COMPLEX, 0.3, color, 1)
if int(classid)==0:
self.count_chuoichin+=1
self.sr.write(chuoichin\\n.encode())
elif int(classid)==1:
self.count_chuoixanh+=1
self.sr.write(chuoixanh\\n.encode())
elif int(classid)==2:
self.count_taodo+=1
self.sr.write(taodo\\n.encode())
elif int(classid)==3:
self.count_taoxanh+=1
self.sr.write(taoxanh\\n.encode())
with open(taodo.txt,a) as f :
f.write(str(self.count_chuoixanh)+\\t+str(self.count_taodo)+\\t+str(self.count_chuoichi
n)+\\t+str(self.count_taoxanh)+\\n)

return classes

65

Code Camera:
import cv2
from base_camera import BaseCamera
import sys
sys.path.append(..)
from yolov4 import *
class Camera(BaseCamera):
def __init__(self):
super().__init__()

def frames(self):
camera = cv2.VideoCapture(/dev/video0, cv2.CAP_V4L)
model=Detection()
if not camera.isOpened():
raise RuntimeError(Could not start camera.)
count=0
while True:
# read current frame
_, frame = camera.read()
count+=1
frame=cv2.resize(frame,(640,480))
# Them count vao day
if count%15==0:
classes=model.detect(frame)
print(classes)
yield frame
if __name__==__main__:
Cam=Camera()
while True :
66

frame=Cam.get_frame()
cv2.imshow(camera,frame)
if cv2.waitKey(1) &amp 0xff==ord(q):
break

67