مدل پیاده سازی شده برای شناسایی پهپاد در تصویر و رسم bbox آموزش دیده است و قابلیت اجرا بر روی cpu و gpu و multi-gpu را دارد.
مدل پیاده سازی شده بر اساس یادگیری عمیق و با استفاده از معماری r-cnn و با شبکه ی keras-retinanet پیاده سازی شده است.
پیش نیازهای پروژه و ابزارهای لازم در فایل requirement.txt موجود است و با استفاده از دستور زیر میتوان آنها را نصب کرد.
pip install -r requirements.txt
برای شناسایی نیاز است مدل آموزش دیده در فولدر Trained-Model قرار داده شود که در حال حاضر مدل آموزش دیده موجود است در صورت آموزش مجدد مدل باید فایل جدید با این فایل جایگزین شود
نکته: برای اجرای مدل بر روی gpu باید در فایل requirement.txt خط tensorflow به tensorflow-gpu تغییر کند.
با استفاده از رابط کاربری زیر میتوان از مدل برای شناسایی پهپاد در تصاویر استفاده کرد.
from core import Core
c = Core()
image_filename = c.current_path + "/DataSets/Drones/testImages/351.jpg"
image = c.load_image_by_path(image_filename)
drawing_image = c.get_drawing_image(image)
processed_image, scale = c.pre_process_image(image)
c.set_model(c.get_model())
boxes, scores, labels = c.predict_with_graph_loaded_model(processed_image, scale)
detections = c.draw_boxes_in_image(drawing_image, boxes, scores)
c.visualize(drawing_image)