Commit graph

11 commits

Author SHA1 Message Date
rht 81234be7da Move the sigmoid activation to the model itself
Former-commit-id: e3f8ca7b1ac7c5e9694637a81be260e9b48973b9
2018-11-10 23:11:52 +00:00
milesial f2d88f98a7 Switch net to eval mode in predict.py
Former-commit-id: 3af421f3060c30b19ee38bc6e00db2593e53ad62
2018-09-26 08:59:49 +02:00
Chang Jo Kim 179b08e6a3 Fixed argument actions
Former-commit-id: 7f855d53668a576236b16670fd052d8caf816d80
2018-07-10 11:19:03 +09:00
milesial bc465b2d91 Fixed some details on predict + requirements
Former-commit-id: d04263be694bba9dc1cfeb96389a3ad2a4015841
2018-06-23 19:57:53 +02:00
ZiyuanTonyZhang ff85f85dcc fix scaling issue by size change from net, i.e. Issues - error in predict.py #12
Former-commit-id: 4af6e54bd4e1c642282fe512d97aa5a09ce16fca
2018-06-17 02:31:42 -04:00
milesial 02e2314149 Migration to PyTorch 0.4, code cleanup
Former-commit-id: c981801ccc3b74047e94c76e67c4ff1f3097226c
2018-06-08 19:27:32 +02:00
milesial 8008b77af6 Removed unused function and general cleanup
Former-commit-id: c34a455f1722e0b899e9e92c7766b83a9a641980
2018-04-09 05:15:24 +02:00
milesial 2d427db832 Cleaned code, added image to README.md
Former-commit-id: 3acf1ff8dadb74e95786fb6ddcf1a90de63f5079
2017-11-30 07:47:43 +01:00
milesial 8b614c3e31 Modified to take any image size (with even width, height > width/2)
Former-commit-id: 2751e6a3df45c1527376a4697d3804d683095d83
2017-11-30 07:19:52 +01:00
milesial 7ea54febec Added CLI for predict, cleaned up code, updated README
Former-commit-id: 77555ccc0925a8fba796ce7e42843d95b6e9dce0
2017-11-30 06:45:19 +01:00
milesial 74ee006a7a Added simple predict + submit script 2017-08-21 18:00:07 +02:00