finder-drone-ai

My project to find anything with a drone! Usecase is with cats, drone will find a cat based on videos of it.

NEW: PLEASE check the summarized presentation video: https://www.youtube.com/watch?v=GN3oGCZcUi4

Extended version with more details: https://youtu.be/N04wvCm5Y_g

Features:

Customized object detector (based on FasterRCNN)
Drone autonomous flying, with visual breaks to fix erratic movements (detects wooden breaks, with Open CV)
Custom image classifier, with an original approach of transfer learning
Federated learning using Raspberry PIs
Real world working demos

Requirements

python main.py --save_session --cat any --mission fffflbbbblfffflbbbbl

Then press T for the drone to take off, and S to start auto pilot mode (will find the cat)

python main.py --save_session --cat <name_of_the_cat>
# For example:
python main.py --save_session --cat lily --mission fffflbbbblfffflbbbbl

and the drone will find Lily (the cat)

For further instructions, notebooks, etc and explanation please check the videos linked above.

Name		Name	Last commit message	Last commit date
Latest commit History 93 Commits
dataset		dataset
globals		globals
images		images
lib		lib
models		models
output		output
pyimagesearch		pyimagesearch
tools		tools
utils		utils
videos		videos
.gitignore		.gitignore
README.md		README.md
acknowledgments.md		acknowledgments.md
create_dataset.py		create_dataset.py
main.py		main.py
mission_utils.test.py		mission_utils.test.py
setup.txt		setup.txt
test_classifier_frame.py		test_classifier_frame.py
test_detector_frame.py		test_detector_frame.py
test_inference.py		test_inference.py
test_video.py		test_video.py
test_video_cat.py		test_video_cat.py
train-4cats.ipynb		train-4cats.ipynb
train-fed-mnist-tensors.ipynb		train-fed-mnist-tensors.ipynb
train-fed-mnist.ipynb		train-fed-mnist.ipynb
train.ipynb		train.ipynb
train_federated.ipynb		train_federated.ipynb
train_federated_rpi.ipynb		train_federated_rpi.ipynb
train_test.py		train_test.py