/basic-yolo-keras

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).

Primary LanguageJupyter NotebookMIT LicenseMIT

YOLOv2 in Keras and Applications

這個repo包含在Keras中使用Tensorflow後端的YOLOv2演算方的實現。它支持訓練YOLOv2網絡與不同的網絡結構,如MobileNet和InceptionV3。

由於原始的專案experiencor/basic-yolo-keras主要以英文來說明演算法的實現與再訓練, 對於一些剛入門深度學習的學習者來說,有一些不容易理解與入手的情況。

因此在這個專案的方向在於文件說明的中文化以外,同時也會以一些便於使用與學習的角度進行源碼的調整與修正。

0. Requirement

python 3.5

keras >= 2.0.8

imgaug

授權

詳細授權宣告 MIT授權.