Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Theano Support #47

Open
wants to merge 23 commits into
base: master
Choose a base branch
from
Open

Theano Support #47

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
70303be
work on removing tensorflow deps
Jun 24, 2017
cdb76f4
add custom layer for space2depth
Jun 24, 2017
ce7f6bc
update to space_to_depth layer
Jun 24, 2017
3c6ceb5
finalize space_to_depth layer
Jun 24, 2017
5a71cf4
try to make stuff work
Jun 24, 2017
a8057cc
finish making yolo theano-friendly
Jun 26, 2017
ef1909a
Update retrain_yolo.py
Jun 26, 2017
0d32e62
make evaluation functions work without keras, update test_yolo.py
Jun 26, 2017
d4dbcd9
Merge branch 'keras_backend' of https://github.com/shadySource/YAD2K …
Jun 26, 2017
bb67609
update training process, fix retrain_yolo.draw() to work with univers…
Jul 10, 2017
0793a76
uncomment training in retrain_yolo.py
Jul 11, 2017
32628e2
fix yad2k weights ordering for theano
Jul 15, 2017
684a1a9
add and use yolo_head_np()
Jul 15, 2017
a952c4f
fix SpaceToDepth fully convolutional bug
Jul 16, 2017
8d942ed
allow all yad2k yolo models with retrain_yolo, save model layout with…
Jul 16, 2017
678f3cf
in retrain_yolo.create_model(), allow variable number of frozen layers
Jul 16, 2017
5550008
fix previous commit
Jul 16, 2017
6dcbcbe
fix yolo_head_np() include in retrain_yolo.py
Jul 16, 2017
6722264
in retrain_yolo.process_data, allow for using datasets with different…
Jul 16, 2017
eaa9ed6
allow for variable train image size in retrain_yolo.get_detector_mask…
Jul 16, 2017
f888cd3
fix yad2k.utils.draw_boxes() for small images
Jul 16, 2017
0c54569
change draw default
Jul 16, 2017
42dbc33
add CUSTOM_DICT to keras_yolo.py to enable simpler loading
Jul 18, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
images/*
images/out/*
model_data/*_anchors.txt
model_data/yolo.json

# Tensorboard training logs
logs/
Expand Down
Loading