diff --git a/Pipfile b/Pipfile new file mode 100644 index 0000000..4764d88 --- /dev/null +++ b/Pipfile @@ -0,0 +1,47 @@ +[[source]] + +url = "https://pypi.python.org/simple" +verify_ssl = true +name = "pypi" + + +[packages] + +Keras = "==2.1.5" +tensorflow = "==1.6.0" +"h5py" = "==2.7.1" +scikit-image = "==0.13.1" +scikit-learn = "==0.19.1" +sk-video = "==1.1.10" +tqdm = "==4.19.6" +coloredlogs = "==9.0" +absl-py = "==0.1.11" +astor = "==0.6.2" +bleach = "==1.5.0" +Cycler = "==0.10.0" +decorator = "==4.2.1" +gast = "==0.2.0" +grpcio = "==1.10.0" +"html5lib" = "==0.9999999" +humanfriendly = "==4.8" +kiwisolver = "==1.0.1" +Markdown = "==2.6.11" +matplotlib = "==2.2.0" +networkx = "==2.1" +numpy = "==1.14.1" +Pillow = "==5.0.0" +protobuf = "==3.5.2" +pyparsing = "==2.2.0" +python-dateutil = "==2.6.1" +pytz = "==2018.3" +PyWavelets = "==0.5.2" +PyYAML = "==3.12" +scipy = "==1.0.0" +six = "==1.11.0" +tensorboard = "==1.6.0" +termcolor = "==1.1.0" +Werkzeug = "==0.14.1" + + +[dev-packages] + diff --git a/Pipfile.lock b/Pipfile.lock new file mode 100644 index 0000000..de46844 --- /dev/null +++ b/Pipfile.lock @@ -0,0 +1,572 @@ +{ + "_meta": { + "hash": { + "sha256": "1e445b8f1ffe3f35f3ef67114b7c9567bc0226bec66033fc92da171fa2d281f8" + }, + "pipfile-spec": 6, + "requires": {}, + "sources": [ + { + "name": "pypi", + "url": "https://pypi.python.org/simple", + "verify_ssl": true + } + ] + }, + "default": { + "absl-py": { + "hashes": [ + "sha256:ba6dc5510108fdfcf7176f8b38a5cc2482f2e902eb7211cbe6666998ea58daf9" + ], + "index": "pypi", + "version": "==0.1.11" + }, + "astor": { + "hashes": [ + "sha256:64c805f1ad6fbc505633416b6174fc23796eb164f371a7dc1f3951ea30560fb5", + "sha256:ff6d2e2962d834acb125cc4dcc80c54a8c17c253f4cc9d9c43b5102a560bb75d" + ], + "index": "pypi", + "version": "==0.6.2" + }, + "bleach": { + "hashes": [ + "sha256:978e758599b54cd3caa2e160d74102879b230ea8dc93871d0783721eef58bc65", + "sha256:e67f46adcec78dbc3c04462f3aba3213a673d5652eba2609ed1ef15492a44b8d" + ], + "index": "pypi", + "version": "==1.5.0" + }, + "coloredlogs": { + "hashes": [ + "sha256:6bd7ceac109c3f2e138db8578396664b1067f32aca55d3280a57dbf05f1ada6c", + "sha256:e3b19320bd21bde506444601a71397cf5215f040df06503013697c6261b05de9" + ], + "index": "pypi", + "version": "==9.0" + }, + "cycler": { + "hashes": [ + "sha256:1d8a5ae1ff6c5cf9b93e8811e581232ad8920aeec647c37316ceac982b08cb2d", + "sha256:cd7b2d1018258d7247a71425e9f26463dfb444d411c39569972f4ce586b0c9d8" + ], + "index": "pypi", + "version": "==0.10.0" + }, + "decorator": { + "hashes": [ + "sha256:7d46dd9f3ea1cf5f06ee0e4e1277ae618cf48dfb10ada7c8427cd46c42702a0e", + "sha256:94d1d8905f5010d74bbbd86c30471255661a14187c45f8d7f3e5aa8540fdb2e5" + ], + "index": "pypi", + "version": "==4.2.1" + }, + "gast": { + "hashes": [ + "sha256:7068908321ecd2774f145193c4b34a11305bd104b4551b09273dfd1d6a374930" + ], + "index": "pypi", + "version": "==0.2.0" + }, + "grpcio": { + "hashes": [ + "sha256:022dc2a6d1537a5a16af4ccc3355ad7b512f9c627a1d5d579cd7c18830378bb3", + "sha256:025a591606b0aca13bec3e019d6acec01a39421f01b915b98a3a93ea0a53b412", + "sha256:03265472d39bf26f124c3ef68446f7873c8260893e6ae65b323a5b51ed52e580", + "sha256:10efe2e016c3ca7a370771ffcf1de9baa3456d4bccefde0f4ce3be091d871c8f", + "sha256:14bca481188c3f19135012aaff9abefa4e15529c7e1aca3084183d78094d06bd", + "sha256:17240d672b5c1c9ff22e52236c1870413b7fb5af762b97ce5a747a55e0a57e98", + "sha256:1bc36e512741f82c1d73f42df536aa2ab75d840f0d35c149b5d0bee1aed16862", + "sha256:224c513fbe0c3ca546870e5c21b08a8a56cd25795b76b3192ee9702a3344764b", + "sha256:2894466c499d9752e0d49ee8adc5ee12c676d86211fc1b292bf713cc7cfe9853", + "sha256:435b3bab2e34814666854eec203c77b169df1cd56cf22fe449cf5510af416e7d", + "sha256:4765600467d7cdb8f62a591d4427ddbeefcf4dbbe46e2f1b10af555e815ecbcb", + "sha256:4fa658a7e1ba5727ca066b1c8bb64c6befb98f2b8007f04a16c7c84555bf11b9", + "sha256:87e52924a99ac5935a468b3fe49c4b0090bd9b05470b55ed1192308791e6d332", + "sha256:88afda198adb0a9da52a66152062027a57877b46f59ffcf55acc3cbfaff77160", + "sha256:982439a872d41f969724efc139e0416ba45e0d7446e9a41fd2ebe19351adff9a", + "sha256:a1bc37c9910d0fbf4d9e80d5822f92c6e01e28dd1eb01323636ed19666b537cb", + "sha256:aa473b8276de39eeccc4ad6cbb7fd7feab0868180d72c0c93226033c79fa69b7", + "sha256:ae82bf2f7ceac6ba956e816120b4f66bda035571350e46b61bbdde1808aed1dd", + "sha256:b56e4f355c2499bb0bf8f8f4d0362b618b06afdfd2c10722710596dc7e295c6c", + "sha256:d2accc8e354f0ed5b337865260a78b3c6851d2fe3c0e1b025d437122cc15dd31", + "sha256:d410835e7554d064c2d99cfa0dd393ffbb0ccf52145ab51c725a8472ed254a3c", + "sha256:d9e3105f6de6cb759b028702bdd21cb36d27e010227669e43c675b9957a3c180", + "sha256:da306c80d69801a3e4115c448ed4ad481957d723ec1e00b99497c6661573c3e5", + "sha256:e579e4124d2a0931ce39639c60e0711918d6659b933eb97e67f60f84666ea488", + "sha256:e86639989c03831912fd9924beda26f6e9ffcc267656cea035bde9d88cf793b2", + "sha256:ea9564f58144e2f07995d57fb8e636be5efb084cd59c8651391ada2bb75dc0ff", + "sha256:f4a38071dd27f140cfe774f56aecdf0e33de926c21289cc9c7521ce8dd91fc1c" + ], + "index": "pypi", + "version": "==1.10.0" + }, + "h5py": { + "hashes": [ + "sha256:07ddea6bb649a257fc57ccae359a36d691b2ef8b9617971ae7d6f74ef6f67cad", + "sha256:180a688311e826ff6ae6d3bda9b5c292b90b28787525ddfcb10a29d5ddcae2cc", + "sha256:1be9cd57e74b24f836d0d2c34ae376ff2df704f40aa8815aa9113b5a860d467f", + "sha256:1fad9aa32835230de77b31edd6980b7c202de7bb7d8384d1bcb47b5dd32c8c7c", + "sha256:2258fca3533a3276fd86e9196326786f408a95748ac707c010fff265edf60342", + "sha256:2b91c9117f2e7a2ef924bec41ac77e57567bec6731773373bf78eb4387b39a2a", + "sha256:2ccb4f405059314829ebad1859d2c68e133a9d13ca7c3cc7a298a76a438fd09c", + "sha256:2d137a1b2f529e58886b5865f6dec51cd96ea0671dd84cebc6dba5cd8c7d0a75", + "sha256:40dd37cbf24ca3b935a8d6eb8960ec5d0381219f82317bdc40aa9e08b3fcc143", + "sha256:478efa37b84a56061af5fcd286678331e873e216f6c5987cd31f9666edc2f157", + "sha256:4a6e6cd8668fa453864f4f9e243460dcc2d41e79d14516b84f4ba74ebcc5b222", + "sha256:52204972a02032d6a427addd37a24a22a2b97d4bce0850c84a6995db9c91926c", + "sha256:537a60879485e5ce484ab4350c7bd8b3da4b531f9f82ef0a18780beabde98c90", + "sha256:562045c57a2e47aca9c716ac8cd64448d4897c0f5fe456ab5a34b17c8b3907cb", + "sha256:66609c48f8841357ced4291b7c9009518bb6e6fec449d91eb46aa417b6f5f4cf", + "sha256:9d9fb861e10735c5c710fe18f34c69e470cf161a4ba38717b7dde21de2d33760", + "sha256:9e0537058efea7547d976f9c00067f7193727bb41ce6b4733c52de35beaa46f5", + "sha256:a314e5e98037ece52ad0b88b4e0d788ca554935268f3e9d293ca9bcd18611b42", + "sha256:b7e1c42367513108c3615cf1a24a9d366fd93eb9d2d92085bafb3011b785e8a9", + "sha256:bb990d8663dbeee22ce44135ffd65ab38bd23d6a689722a653cfbf2d18d46688", + "sha256:c050791989cd9979fe57a770d4e323b2e67ef95800e89e7dc6ad3652b8ccd86f", + "sha256:e1bfcfa2c425dc0f637d4edd858b94e400bbb5746dba324ace124d55fc21d3df", + "sha256:e78f09a44fc9256b84c9df98edf7b6ead3b3da2e12bf2d1e00384960a6a78a1a" + ], + "index": "pypi", + "version": "==2.7.1" + }, + "html5lib": { + "hashes": [ + "sha256:2612a191a8d5842bfa057e41ba50bbb9dcb722419d2408c78cff4758d0754868" + ], + "index": "pypi", + "version": "==0.9999999" + }, + "humanfriendly": { + "hashes": [ + "sha256:587b16ce804bec8e3cbb8c420decea051b38e3d895272b2c1e38fc69b4286b1c", + "sha256:d0e74171b87318a94b99520e4f0c5651e944b5f11d696c46be3330bb82b85300" + ], + "index": "pypi", + "version": "==4.8" + }, + "keras": { + "hashes": [ + "sha256:55e4598f6ff2a374a8abd8e7a4c65be95ac5d3a6ee81b08d7d2549f8c3e1d8d5", + "sha256:907ad29add1fff27342a9f4fe3e60003d450d3af41a38f22f629c7736fc8399d" + ], + "index": "pypi", + "version": "==2.1.5" + }, + "kiwisolver": { + "hashes": [ + "sha256:0ee4ed8b3ae8f5f712b0aa9ebd2858b5b232f1b9a96b0943dceb34df2a223bc3", + "sha256:0f7f532f3c94e99545a29f4c3f05637f4d2713e7fd91b4dd8abfc18340b86cd5", + "sha256:1a078f5dd7e99317098f0e0d490257fd0349d79363e8c923d5bb76428f318421", + "sha256:1aa0b55a0eb1bd3fa82e704f44fb8f16e26702af1a073cc5030eea399e617b56", + "sha256:2874060b91e131ceeff00574b7c2140749c9355817a4ed498e82a4ffa308ecbc", + "sha256:379d97783ba8d2934d52221c833407f20ca287b36d949b4bba6c75274bcf6363", + "sha256:3b791ddf2aefc56382aadc26ea5b352e86a2921e4e85c31c1f770f527eb06ce4", + "sha256:4329008a167fac233e398e8a600d1b91539dc33c5a3eadee84c0d4b04d4494fa", + "sha256:45813e0873bbb679334a161b28cb9606d9665e70561fd6caa8863e279b5e464b", + "sha256:53a5b27e6b5717bdc0125338a822605084054c80f382051fb945d2c0e6899a20", + "sha256:66f82819ff47fa67a11540da96966fb9245504b7f496034f534b81cacf333861", + "sha256:79e5fe3ccd5144ae80777e12973027bd2f4f5e3ae8eb286cabe787bed9780138", + "sha256:8b6a7b596ce1d2a6d93c3562f1178ebd3b7bb445b3b0dd33b09f9255e312a965", + "sha256:9576cb63897fbfa69df60f994082c3f4b8e6adb49cccb60efb2a80a208e6f996", + "sha256:95a25d9f3449046ecbe9065be8f8380c03c56081bc5d41fe0fb964aaa30b2195", + "sha256:aaec1cfd94f4f3e9a25e144d5b0ed1eb8a9596ec36d7318a504d813412563a85", + "sha256:acb673eecbae089ea3be3dcf75bfe45fc8d4dcdc951e27d8691887963cf421c7", + "sha256:b15bc8d2c2848a4a7c04f76c9b3dc3561e95d4dabc6b4f24bfabe5fd81a0b14f", + "sha256:b1c240d565e977d80c0083404c01e4d59c5772c977fae2c483f100567f50847b", + "sha256:ce3be5d520b4d2c3e5eeb4cd2ef62b9b9ab8ac6b6fedbaa0e39cdb6f50644278", + "sha256:e0f910f84b35c36a3513b96d816e6442ae138862257ae18a0019d2fc67b041dc", + "sha256:ea36e19ac0a483eea239320aef0bd40702404ff8c7e42179a2d9d36c5afcb55c", + "sha256:f923406e6b32c86309261b8195e24e18b6a8801df0cfc7814ac44017bfcb3939" + ], + "index": "pypi", + "version": "==1.0.1" + }, + "markdown": { + "hashes": [ + "sha256:9ba587db9daee7ec761cfc656272be6aabe2ed300fece21208e4aab2e457bc8f", + "sha256:a856869c7ff079ad84a3e19cd87a64998350c2b94e9e08e44270faef33400f81" + ], + "index": "pypi", + "version": "==2.6.11" + }, + "matplotlib": { + "hashes": [ + "sha256:08de895360bd5dc489514b395ff5da153931939145ae4608afb78799dc592dc0", + "sha256:3947b26e4ba51276df41167614a0fde9fba43740c7999fbd2b5de34fd75119cc", + "sha256:3ca7a2c17b5c4c21177f764abca197ff9ac0ac68352e24718d05929c8592ed0d", + "sha256:504c3a891fcfdb48e94983c50a5b8b626f25000d407ea2d9ab9ed5691762fbe8", + "sha256:5837e3dd76d43aa2fe0a90c79f6c1d4d38c205550e8ae685bda176af62c86bbd", + "sha256:6cbe2b9cdb61fdea331e1e1feebeb1d03c49cb84af43957767e5471b99ddb822", + "sha256:8917e89672b7668861f6db45561bd5fc1ecd240089ab0723e4ca96bd1e1f44ce", + "sha256:8df8c75938193cda5a041762fcc96f47658a6125fae07dc0678502dba3b62f4c", + "sha256:9819535c069d7207dd55f3992875e96a3ea6a787ddc37afea00cc95bba4125ee", + "sha256:b91ee711c7e6e4c43f106f223e9eac50e98f35153813f5fd70fd592be3616ade", + "sha256:c33cf0f2a2064b3ae60f9fb806a1da5addaf6be66ec3d106794955e3888d8442", + "sha256:c7d2524971ba0b9da1b04a9054749f8654298573243db6eb762d7f58daece1b4", + "sha256:d65879e7ddce62fc2e6339216a0c0273b3f0e4b2401f7edb512efcbd72d13e9b", + "sha256:d744e56eb6742a4a3a9dbbb22c71bd0d6262543a2817fb35c9ed21ff4d06e4ce", + "sha256:da2ccf36a7f09d07b64b8bb708f1389f8493a5326a12ba959483b9530c64fd23", + "sha256:f652a3f23ea347a57eb6743ba83cd91ac347bed08fb82acde0e015a9afacbbd9", + "sha256:f6c0fd70adbe1e41d11b3665409566a512ee037268654e558d79f669a6dd183f", + "sha256:fe816662169026f6d13063be05f7ccadb11d80e737351b778ceb3fa17ef23002" + ], + "index": "pypi", + "version": "==2.2.0" + }, + "networkx": { + "hashes": [ + "sha256:64272ca418972b70a196cb15d9c85a5a6041f09a2f32e0d30c0255f25d458bb1" + ], + "index": "pypi", + "version": "==2.1" + }, + "numpy": { + "hashes": [ + "sha256:12cf4b27039b88e407ad66894d99a957ef60fea0eeb442026af325add2ab264d", + "sha256:2da8dff91d489fea3e20155d41f4cd680de7d01d9a89fdd0ebb1bee6e72d3800", + "sha256:377def0873bbb1fbdedb14b3275b10a29b1b55619a3f7f775c4e7f9ce2461b9c", + "sha256:3d7ddd5bdfb12ec9668edf1aa49a4a3eddb0db4661b57ea431477eb9a2468894", + "sha256:49880b47d7272f902946dd995f346842c95fe275e2deb3082ef0495f0c718a69", + "sha256:4e2fc841c8c642f7fd44591ef856ca409cedba6aea27928df34004c533839eee", + "sha256:6b1011ffc87d7e2b1b7bcc6dc21bdf177163658746ef778dcd21bf0516b9126c", + "sha256:6b8c2daacbbffc83b4a2ba83a61aa3ce60c66340b07b962bd27b6c6bb175bee1", + "sha256:788e1757f8e409cd805a7cd82993cd9252fa19e334758a4c6eb5a8b334abb084", + "sha256:7d4c549e41507db4f04ec7cfab5597de8acf7871b16c9cf64cebcb9d39031ca6", + "sha256:81b9d8f6450e752bd82e7d9618fa053df8db1725747880e76fb09710b57f78d0", + "sha256:89b9419019c47ec87cf4cfca77d85da4611cc0be636ec87b5290346490b98450", + "sha256:9501c9ccd081977ca5579a3ec4009d6baff6bacb04bf07214aade3324734195a", + "sha256:9b762e78739b6e021124adbea07611682db99cd3fca7f3c3a8b98b8f74ea5699", + "sha256:a1f5173df8190ef9c6235d260d70ca70c6fb029683ceb66e244c5cc6e335947a", + "sha256:a3d5dd437112292c707e54f47141be2f1100221242f07eda7bd8477f3ddc2252", + "sha256:a8bc80f69570e11967763636db9b24c1e3e3689881d10ae793cec74cf7a627b6", + "sha256:b803306c4c201e7dcda0ce1b9a9c87f61a7c7ce43de2c60c8e56147b76849a1a", + "sha256:c8000a6cbc5140629be8c038c9c9cdb3a1c85ff90bd4180ec99f0f0c73050b5e", + "sha256:e2335d56d2fd9fc4e3a3f2d3148aafec4962682375f429f05c45a64dacf19436", + "sha256:e5ade7a69dccbd99c4fdbb95b6d091d941e62ffa588b0ed8fb0a2854118fef3f", + "sha256:e8522cad377cc2ef20fe13aae742cc265172910c98e8a0d6014b1a8d564019e2", + "sha256:fa0944650d5d3fb95869eaacd8eedbd2d83610c85e271bd9d3495ffa9bc4dc9c" + ], + "index": "pypi", + "version": "==1.14.1" + }, + "pillow": { + "hashes": [ + "sha256:0013f590a8f260df60bcfd65db19d18efc04e7f046c3c82a40e2e2b3292a937c", + "sha256:0b899ee80920bb533f26581af9b4660bc12aff4562555afe74e429101ebf3c94", + "sha256:12f29d6c23424f704c66b5b68c02fe0b571504459605cfe36ab8158359b0e1bb", + "sha256:135e9aa65150c53f7db85bf2bebb8a0e1a48ea850e80cf66e16dd04fa09d309c", + "sha256:153ec6f18f7b61641e0e6e502acfaf4a06c9aba2ea11c0b4b3578ea9f13a4a4a", + "sha256:17fe25efc785194d48c38fad85dce470013ba19d2fb66639e149f14bccf1327f", + "sha256:1912b7230459fd53682dae32b83cbd8e5d642ba36d4be18566f00a9c063aa13d", + "sha256:1a5b93084e01328a1cb1ecdad99d11d75e881e89a95f88d85b523646553b36c2", + "sha256:25193f934d37d836a6b1f4c062ce574a96cbca7c6d9dc8ddfbbac7f9c54deaa4", + "sha256:2c042352b430d678db50c78c5214e19638eff8b688941271da2de21fd298dfe5", + "sha256:2e818dbe445e86fc6c266973fe540c35125c42eb2cf13a6095e9adaa89c0deb5", + "sha256:2fcde9954c8882d1c7f93bb828caa34a4c5e3ee69dbc7895dc8652ad972b455a", + "sha256:35f7d998b8e82fb3fb51ff88b30485eb81cd7dd56ec7e1a8deba23eb88532d44", + "sha256:37cc0339abfa9e295c75d9a7f227d35cb44716feb95057f9449c4a9e9a17daf7", + "sha256:43334f9581cd067945b8898cef9eb5714ee4883f8de0304c011f1dbdb1d4e2aa", + "sha256:4bd4a71501b6d51db4abc07e1f43f5a6fed0a1a9583cca0b401d6af50284b0db", + "sha256:57aa6198ba8acba1313c3b743e267d821a60cac77e6026caf0b55ca58d3d23be", + "sha256:5b0d657460d9f3615876fec6306e97ca15a471f6169b622d76a47e270998acf1", + "sha256:5cd36804f9f06a914a883fe682df5711d16d7b4f44d43189c5f013e7cd91e149", + "sha256:6977cf073d83358b34f93abf5c1f1193b88675fe0e4441e0e28318bc3dcba7a0", + "sha256:718ec7a122b28d64afc5fbc3a9b99bb0545ef511373cac06fe7624520e82cb20", + "sha256:7dfbefdb3fb911ca9faed307bf309861e9995e36cca6b761c7ba6d9b77a9744a", + "sha256:801cca8923508311bf5d6d0f7da5362552e8208ebd8ec0d7b9f2cd2ff5705734", + "sha256:82b172e3264e62372c01b5b009b5b1a02fbb9276cbe5cc57ab00a6d6e5ed9a18", + "sha256:82d1ff571489765df2816785d532e243bde213752156c227fca595723ec5ff42", + "sha256:8580fc58074a16b749905b26cf8363f7b628dd167ba0130f5382cdc91c86b509", + "sha256:931030d1d6282b7900e6b0a7ff9ecdb503b5e1e6781800dab2b71a9f39405bff", + "sha256:9525cd680a6f9e80c6c0af03cf973e6505c59f60b4745f682cd1a449e54b31bb", + "sha256:a224651a81e45ef4f1d0164e256c5f6b4abb49f2ae8f22ba2f3a9d0ff338e608", + "sha256:a370d1c570f1d72e877099651e752332444b1c5009381f043c9da5fd47f3ebae", + "sha256:b1d33c63a55d0d85df0ad02b2c16158fb4d8153afa7b908f1a67330fac694cd6", + "sha256:b2240f298482f823576f397bb9f32ea913ad9456c526e141bc6f0a022b37a3e8", + "sha256:b85f703c2ffe539313e39ce0676bed0f355cec45a16e58c9ab7417445843047c", + "sha256:b9f63451084a718eccdeb1e382768c94647915653af4d6019f64560d9e98642b", + "sha256:c793dfaa130847ccff958492b76ae8b9304e60b8a79a92962cb19e368276a22b", + "sha256:d60c1625b108432ace8b1fa1a584017e5efa73f107d0f493c7f39c79bebf1d41", + "sha256:dc4b018d5c9b636f7546583c5591b9ea00c328c3e5871992ef5b95bac353f097", + "sha256:ddd16ab250b4fc97db1c47407e78c25216a75c29d29d10ad37e51b7a2ec7b2c3", + "sha256:e126ff4fed71e78333840c07279e1617f63cfca76d63ad5b27d65a7277206a3d", + "sha256:f8d49be8c282df8d2e1ab6ab53ab8abd859b1fa6fed384457ee85c9eff64ef97", + "sha256:fcf64c91fd44485100a2965d23bb0e227d093e91f7e776c5ca3b32574766eb56" + ], + "index": "pypi", + "version": "==5.0.0" + }, + "protobuf": { + "hashes": [ + "sha256:09879a295fd7234e523b62066223b128c5a8a88f682e3aff62fb115e4a0d8be0", + "sha256:14813a3421ff0144e8d4e81ed83a3fbe350d8d85cbe480bf2e81cf45e8083e0d", + "sha256:18a4a387e8378dbbd53ebe9cc925ea2fe2a7b98c497833ea345803cb53b885d9", + "sha256:24c1cc840b4832a909bbeac664fd8f878cf72b8ab97bfe4fb82a156c3f1f0e15", + "sha256:40c943a8ffb3501164da1d2b537ad2e33d08daf81fbb3e9073bf291726a24467", + "sha256:41e916354265d2f54b95e454305c98f90bb30fafb817119540753e67f193de57", + "sha256:59610aeb5ade675106dca26c771814a1aa63bf2b3780584853e3dd447ed5c52f", + "sha256:59ff8a204aa2ef98d6c25c2adffb13dda81bb4ac6ffb0829c92e801241b6477b", + "sha256:64a3600d2a531d7c516c371efa431035ce501ab8425dcc8bdb99eddf5a4d34c9", + "sha256:6e1c0972462ce9dc4d2860d533487b39f89de00b3f30b99c31a6b3e8fbf8b787", + "sha256:75e1a7b12248a98b620ffbda3e41767aa2ae57c7cc553a12407a48c44f58f2e7", + "sha256:84ed523853c82c76dd1dfd15f31de2d66fa7cb22a48aa42dbc32465868d7e4af", + "sha256:87908d494be2b46a55de5e55ca11d9a2508b59b035c1b0549c3b692a77f57a7b", + "sha256:88c7958dad426920a43af58c5805d2de860a33f82d47f5a102af25f2788682c7", + "sha256:8ba58356fc40ed7749c73eeae3d86f6a9e756ba1ae5f5833990b237b7d61ba09", + "sha256:94d159e2bbbe4df1b5f0715965e284f2156ce127a7d521a3dcbdd38e945bc4c0", + "sha256:c4d531e745168c16fc7abff12922c491d34f4063c1b49fe5417b72be869f5df6", + "sha256:e457146bb9f997736460b10b2f2a9284603db4bbd60c8c431b5b4b309efbe036", + "sha256:e774cd03628c0b2f850a09a8c005fe6113f97e37f6df07a7b20221dc1ee4efd3", + "sha256:ec51286554eceebcf169a3a8604861e113d28fc98094dcbedc6067f058478917" + ], + "index": "pypi", + "version": "==3.5.2" + }, + "pyparsing": { + "hashes": [ + "sha256:0832bcf47acd283788593e7a0f542407bd9550a55a8a8435214a1960e04bcb04", + "sha256:281683241b25fe9b80ec9d66017485f6deff1af5cde372469134b56ca8447a07", + "sha256:8f1e18d3fd36c6795bb7e02a39fd05c611ffc2596c1e0d995d34d67630426c18", + "sha256:9e8143a3e15c13713506886badd96ca4b579a87fbdf49e550dbfc057d6cb218e", + "sha256:b8b3117ed9bdf45e14dcc89345ce638ec7e0e29b2b579fa1ecf32ce45ebac8a5", + "sha256:e4d45427c6e20a59bf4f88c639dcc03ce30d193112047f94012102f235853a58", + "sha256:fee43f17a9c4087e7ed1605bd6df994c6173c1e977d7ade7b651292fab2bd010" + ], + "index": "pypi", + "version": "==2.2.0" + }, + "python-dateutil": { + "hashes": [ + "sha256:891c38b2a02f5bb1be3e4793866c8df49c7d19baabf9c1bad62547e0b4866aca", + "sha256:95511bae634d69bc7329ba55e646499a842bc4ec342ad54a8cdb65645a0aad3c" + ], + "index": "pypi", + "version": "==2.6.1" + }, + "pytz": { + "hashes": [ + "sha256:07edfc3d4d2705a20a6e99d97f0c4b61c800b8232dc1c04d87e8554f130148dd", + "sha256:3a47ff71597f821cd84a162e71593004286e5be07a340fd462f0d33a760782b5", + "sha256:410bcd1d6409026fbaa65d9ed33bf6dd8b1e94a499e32168acfc7b332e4095c0", + "sha256:5bd55c744e6feaa4d599a6cbd8228b4f8f9ba96de2c38d56f08e534b3c9edf0d", + "sha256:61242a9abc626379574a166dc0e96a66cd7c3b27fc10868003fa210be4bff1c9", + "sha256:887ab5e5b32e4d0c86efddd3d055c1f363cbaa583beb8da5e22d2fa2f64d51ef", + "sha256:ba18e6a243b3625513d85239b3e49055a2f0318466e0b8a92b8fb8ca7ccdf55f", + "sha256:ed6509d9af298b7995d69a440e2822288f2eca1681b8cce37673dbb10091e5fe", + "sha256:f93ddcdd6342f94cea379c73cddb5724e0d6d0a1c91c9bdef364dc0368ba4fda" + ], + "index": "pypi", + "version": "==2018.3" + }, + "pywavelets": { + "hashes": [ + "sha256:012dfae798fb7f6e522d0c9789ce73b442d2fe98396a6091adfacd1b24d4ae2f", + "sha256:09f42f82ecc1be2f293f83f4e9c55866272cf9f92ac7a3b72705cba7512bd723", + "sha256:15ae04312c007501b316d24de41f8df9c2a0fa7089e3724d584cd7b1cfd8fcae", + "sha256:1b9f0af21aa8b211cda4ff4083478a9fb66ad5a3f2c06300ac84fd9377d95dfc", + "sha256:1d9094b9e5204e446b7ea750b4f5f0bec8d62f2abd740484addde593cfe8c4d5", + "sha256:2e596bbc4e9bc0a544cecdb1c2355ebda73e5a5afc25b20dfa2cded8ef20c94c", + "sha256:30782f2f38d7b89cd6c0562bba70d6cdd14eeac53d9164ee58d2a21714f34047", + "sha256:3cdf80fec5d93ccf5d529e4a481a4a318b14b44624cc8857686e7c682f73e706", + "sha256:4c4aa204cc2cee71f5ed1302caa69a0a5e1d2b36ad56997b7f64e56cf8415a78", + "sha256:50f386b422cd7005b8f69fc20d1502182d80e5b9f7345d5ca4b9ff640364d77d", + "sha256:5bc9546c6f3d1af03d0d8d7a58459115cb526c337317f64f13486e0be96a51a7", + "sha256:5e78b45c8726249df7225da1cf1d08e07ca4d84dd66d49182f23b090f48e21dc", + "sha256:6607067b46ecbac6f6f2ebca83bcc20e51d77c022c6f033f62b9ec173eee5f60", + "sha256:8287961e62f4049491b4ae6d64da71bf98acdbbe593a62c6ceb0be5f06f4d253", + "sha256:9d8729b972943cbd1849a75beb6a87878eb3f9fa9639d027f240d25c4269ae84", + "sha256:b326fd1788beaf11cb14c7c336808f46a01c43e40294070a906eedd865b67d5f", + "sha256:c8795dcabedf97b3ad4978102140c40ab8f22579eb1e7516d251d57a58dd46c1", + "sha256:ce36e2f0648ea1781490b09515363f1f64446b0eac524603e5db5e180113bed9", + "sha256:d69289039d7eb2ebc3275db7bf13d34df0c1bdf51955466a8ef792a309aaf0fe", + "sha256:e6cf3644166884a35fe45c1027cbc76bcdd5c17a74b013f51f54e73a522b1393", + "sha256:eb7f30782ba48cd06b957f6c7728bdcdcc3159b7805bdbd362370d8f5cb81603", + "sha256:ebb3b3f460b1bd5214c49c18ee20971c8efffb9fa482fce0687be1630202cd65", + "sha256:f801fa177f2756da4d7c25ff49f0f09bf56adbdfb1e05582f377948d2faf18de" + ], + "index": "pypi", + "version": "==0.5.2" + }, + "pyyaml": { + "hashes": [ + "sha256:0c507b7f74b3d2dd4d1322ec8a94794927305ab4cebbe89cc47fe5e81541e6e8", + "sha256:16b20e970597e051997d90dc2cddc713a2876c47e3d92d59ee198700c5427736", + "sha256:3262c96a1ca437e7e4763e2843746588a965426550f3797a79fca9c6199c431f", + "sha256:326420cbb492172dec84b0f65c80942de6cedb5233c413dd824483989c000608", + "sha256:4474f8ea030b5127225b8894d626bb66c01cda098d47a2b0d3429b6700af9fd8", + "sha256:592766c6303207a20efc445587778322d7f73b161bd994f227adaa341ba212ab", + "sha256:5ac82e411044fb129bae5cfbeb3ba626acb2af31a8d17d175004b70862a741a7", + "sha256:5f84523c076ad14ff5e6c037fe1c89a7f73a3e04cf0377cb4d017014976433f3", + "sha256:827dc04b8fa7d07c44de11fabbc888e627fa8293b695e0f99cb544fdfa1bf0d1", + "sha256:b4c423ab23291d3945ac61346feeb9a0dc4184999ede5e7c43e1ffb975130ae6", + "sha256:bc6bced57f826ca7cb5125a10b23fd0f2fff3b7c4701d64c439a300ce665fff8", + "sha256:c01b880ec30b5a6e6aa67b09a2fe3fb30473008c85cd6a67359a1b15ed6d83a4", + "sha256:ca233c64c6e40eaa6c66ef97058cdc80e8d0157a443655baa1b2966e812807ca", + "sha256:e863072cdf4c72eebf179342c94e6989c67185842d9997960b3e69290b2fa269" + ], + "index": "pypi", + "version": "==3.12" + }, + "scikit-image": { + "hashes": [ + "sha256:0de04a28b48ec00ad4ba8197fa1e6bd4d382609bf56b5ba23971c6fa3bddeccf", + "sha256:199ba92019c5549783aa629b615394bec7302572b6f5727017a78d3ccc737de9", + "sha256:21c989eea994597a4ccb5b6cb16f6700a0b39d3e9f6c2a83a692c7345b02eb5e", + "sha256:2c316bdec863bdd5eaa5d527a7b82a4e78237816faec46bd178b8ff7e30b5500", + "sha256:353879ddf7483f4621872c49cd9bc8a0ad1c3154ac0670b70799922f4cb899e8", + "sha256:591895cf2620df3aef510c658f93069dc5e482bc244544fe2dbeda8578dcf5c5", + "sha256:5c27b7b7500e3405114c8fda474a5be27823741c0e8e7cdb7142c8da8895f8fb", + "sha256:7dfcd39eb004c2e1595ad6ab1d0dd1e83bc065efe6260376bb4b45df12cfa101", + "sha256:944feb2920ac6a8fc63f05c4abc6811d6fd270b7b5a586cef37cb4ce65534355", + "sha256:9535c1e770f476f8d79992bf632fce01da80f52a0a7e34da56834e7483771afe", + "sha256:9fdef28d7f54411d80d37bbcede324c45abf44ee4b804f25e52c5967dfc2a00c", + "sha256:a01b1d29a67b1721dcc674f88d8f8c3660c16e84a914bf4eef8ba6bf135379d7", + "sha256:b16e80c49bdd764d8e22e9a4edf741913c5b5b6c0918d8c8f266c2b5e385553e", + "sha256:cdb41868c766d4607ded5c158fc195853859a0e3e108ab5804c3acdd3d592d86", + "sha256:d109ffd1e2a314c6cb448d22e8a5fd0b8f39031faeeb38b63fbb104822f06d3a", + "sha256:d5a12eb267cb901de57318d00d3e7514035f10a8b5d6710d7ecce7d106ca0b43", + "sha256:dc2c0a351d1c30448c52df007ef1aa3d2a49dd1ded132e89a217e8ec4afe0f04", + "sha256:e18953a500faed68163e6cea9b274865dff27faee9186fa76a637cb23391908e", + "sha256:f1a06c39888cfdab005cf297d854a204209d7623586546a1364ee996eb41e423" + ], + "index": "pypi", + "version": "==0.13.1" + }, + "scikit-learn": { + "hashes": [ + "sha256:13136c6e4f6b808569f7f59299d439b2cd718f85d72ea14b5b6077d44ebc7d17", + "sha256:370919e3148253fd6552496c33a1e3d78290a336fc8d1b9349d9e9770fae6ec0", + "sha256:3775cca4ce3f94508bb7c8a6b113044b78c16b0a30a5c169ddeb6b9fe57a8a72", + "sha256:42f3c5bd893ed73bf47ccccf04dfb98fae743f397d688bb58c2238c0e6ec15d2", + "sha256:56cfa19c31edf62e6414da0a337efee37a4af488b135640e67238786b9be6ab3", + "sha256:5c9ff456d67ef9094e5ea272fff2be05d399a47fc30c6c8ed653b94bdf787bd1", + "sha256:5ca0ad32ee04abe0d4ba02c8d89d501b4e5e0304bdf4d45c2e9875a735b323a0", + "sha256:5db9e68a384ce80a17fc449d4d5d9b45025fe17cf468429599bf404eccb51049", + "sha256:6e0899953611d0c47c0d49c5950082ab016b38811fced91cd2dcc889dd94f50a", + "sha256:72c194c5092e921d6107a8de8a5adae58c35bbc54e030ba624b6f02fd823bb21", + "sha256:871669cdb5b3481650fe3adff46eb97c455e30ecdc307eaf382ef90d4e2570ab", + "sha256:873245b03361710f47c5410a050dc56ee8ae97b9f8dcc6e3a81521ca2b64ad10", + "sha256:8b17fc29554c5c98d88142f895516a5bec2b6b61daa815e1193a64c868ad53d2", + "sha256:95b155ef6bf829ddfba6026f100ba8e4218b7171ecab97b2163bc9e8d206848f", + "sha256:a21cf8217e31a9e8e32c559246e05e6909981816152406945ae2e3e244dfcc1f", + "sha256:a58746d4f389ea7df1d908dba8b52f709835f91c342f459a3ade5424330c69d1", + "sha256:b2a10e2f9b73de10d8486f7a23549093436062b69139158802910a0f154aa53b", + "sha256:ba3fd442ae1a46830789b3578867daaf2c8409dcca6bf192e30e85beeabbfc2f", + "sha256:ce78bf4d10bd7e28807c36c6d2ab25a9934aaf80906ad987622a5e45627d91a2", + "sha256:d384e6f9a055b7a43492f9d27779adb717eb5dcf78b0603b01d0f070a608d241", + "sha256:d4da369614e55540c7e830ccdd17ab4fe5412ff8e803a4906d3ece393e2e3a63", + "sha256:ddc1eb10138ae93c136cc4b5945d3977f302b5d693592a4731b2805a7d7f2a74", + "sha256:e54a3dd1fe1f8124de90b93c48d120e6da2ea8df29b6895325df01ddc1bd8e26", + "sha256:ee8c3b1898c728b6e5b5659c233f547700a1fea13ce876b6fe7d3434c70cc0e0", + "sha256:f528c4b2bba652cf116f5cccf36f4db95a7f9cbfcd1ee549c4e8d0f8628783b5", + "sha256:f9abae483f4d52acd6f660addb1b67e35dc5748655250af479de2ea6aefc6df0", + "sha256:fdc39e89bd3466befb76dfc0c258d4ccad159df974954a87de3be5759172a067" + ], + "index": "pypi", + "version": "==0.19.1" + }, + "scipy": { + "hashes": [ + "sha256:01c7040a83eb4e020ab729488637dcadef54cb728b035b76668ab92a72515d60", + "sha256:046705c604c6f1d63cad3e89677c0618b7abb40ed09a4c241c671a2d8e5128a9", + "sha256:08041e5336fcd57defcc78650b44b3df652eff3e3a801638d894e50494fb630d", + "sha256:1f58fbd59e8d9652759df0d137832ff2a325ed708c173cba20c86589d811c210", + "sha256:23a7238279ae94e088396b8b05a9795ef598dc79c5cd1adb91ad1ff87c7514fd", + "sha256:3b66d5e40152175bca75cbbfd1eb5c108c50de9ae5625923f1c4f8f51cbe2dea", + "sha256:424500b2fe573d30de6dea927076c01acaadb3efb3d1f40340e8cc37151ccf27", + "sha256:5774adb6047983489bc81edaa72cd132e665e5680f0b2cf8ea28cd3b99e65d39", + "sha256:5964dba6a3c0be226d44d2520de8fb4ba1501768bad57eec687d36d3f53b6254", + "sha256:70e6fc3f2f52c9152f05e27eb9bd8543cb862cacb71f8521a571e4ffb837f450", + "sha256:87ea1f11a0e9ec08c264dc64551d501fa307289460705f6fccd84cbfc7926d10", + "sha256:889602ead28054a15e8c26e1a6b8420d5a4fa777cfeb3ec98cfa52b9f317d153", + "sha256:912499ddb521b7ac6287ac4ccf5f296a83d38996c2d04f43c9e62a91f7b420aa", + "sha256:97123a25216616723083942eb595f47fee18da6b637a88b803de5f078009003c", + "sha256:9bd193686fd837472bdb6425486cb234ed0a4db76b930c141cc8d095ab213c8d", + "sha256:a79b99b8b5af9a63312bd053bbb7bdb7710e6bbb9cc81617f9f6b9b1e49c72f8", + "sha256:a9e479648aab5f36330da94f351ebbfe79acb4e6f5e6ac6aeddc9291eb096839", + "sha256:bf36f3485e7b7291c36330a93bbfd4f5e8db23bbe4ea46c37b2839fef463f4e2", + "sha256:cd23894e1cc6eaa00e6807b6b12e4ca66d5ff092986c9c3eb01e97f24e2d6462", + "sha256:d84df0bc86bbdd49f0a6b6bad5cd62ccb02a3bfe546bf79263de44ae081bcd7b", + "sha256:e3a5673c105eab802fdecb77f102d877352e201df9328698a265b7f57546b34b", + "sha256:fa17be6c66985931d3a391f61a6ba97c902585cf26020aa3eb24604115732d22", + "sha256:ff8b6637d8d2c074ed67f3d57513e62f94747c6f1210f43e60ad3d8e93a424e4" + ], + "index": "pypi", + "version": "==1.0.0" + }, + "six": { + "hashes": [ + "sha256:70e8a77beed4562e7f14fe23a786b54f6296e34344c23bc42f07b15018ff98e9", + "sha256:832dc0e10feb1aa2c68dcc57dbb658f1c7e65b9b61af69048abc87a2db00a0eb" + ], + "index": "pypi", + "version": "==1.11.0" + }, + "sk-video": { + "hashes": [ + "sha256:261aec6423a0188968771e532e6ccdbc53bcbd934d349280784210d9ca9ffe76", + "sha256:fb7125cfa2b31253dcf181d0bf285edd6bb25ffebf61dbd8ba11fadb783609a1" + ], + "index": "pypi", + "version": "==1.1.10" + }, + "tensorboard": { + "hashes": [ + "sha256:835ecbfee0ce505f8365435c23d1c7d6fd527fcad8a4829e4145b9f6f41d0ca2", + "sha256:ab3e4568a277d4d06fc2928e6e5aa2a32ece073ad234a6b7ade08671dbf8f339" + ], + "index": "pypi", + "version": "==1.6.0" + }, + "tensorflow": { + "hashes": [ + "sha256:0b41fb0d1d0be2d495dad67138b5a118f7b206a07b2c96474414109dbfe056a8", + "sha256:17981691a5711dd8b0479b293229f3a22d2f0503d04c2691a2ef66ee50903257", + "sha256:1ba17a5df0c2d3e0000f88479b3ce7a35e05ced3056e4594781e81afb45ac8fd", + "sha256:2f89cf8a05198e7e2415d040b679b4f50ef86c0c3834f4bf86d0b836ecc14a27", + "sha256:638e690a4efc675dbfd0bb98f0c5db1b6b54e9791513fced4a4df80a3db946cb", + "sha256:6d6f8ac26f857ab79c13b594efcb13d5485d1dbfdaa8e54b064ea7b4b28a9cf8", + "sha256:8c504b9c70649d8602592b148944b34207fde8d1f5d61aaf435dab99118a58bc", + "sha256:afb1f05ab3f8ff569b91be5ecb066b926dff1afd029f4f296964317f461268a6", + "sha256:b6b4c52c61f9cea58a93e4c178e8acec157ab2ed8824350ec536f107fa67a491", + "sha256:d0e5daf51cf9a711a0dc95b5bf4a9eeeb1cb485197cb923b8940f2a905850407", + "sha256:d9c628c857ccd9213d75e0747fcad6ef4d6866a79d1a3987d87a5957c3f0f819", + "sha256:eaa4af9eb161af500a5e3c65dcc434d63478801539f51cdd8c01572eda9dfa2d" + ], + "index": "pypi", + "version": "==1.6.0" + }, + "termcolor": { + "hashes": [ + "sha256:1d6d69ce66211143803fbc56652b41d73b4a400a2891d7bf7a1cdf4c02de613b" + ], + "index": "pypi", + "version": "==1.1.0" + }, + "tqdm": { + "hashes": [ + "sha256:5ec0d4442358e55cdb4a0471d04c6c831518fd8837f259db5537d90feab380df", + "sha256:f66468c14ccd011a627734c9b3fd72f20ce16f8faecc47384eb2507af5924fb9" + ], + "index": "pypi", + "version": "==4.19.6" + }, + "werkzeug": { + "hashes": [ + "sha256:c3fd7a7d41976d9f44db327260e263132466836cef6f91512889ed60ad26557c", + "sha256:d5da73735293558eb1651ee2fddc4d0dedcfa06538b8813a2e20011583c9e49b" + ], + "index": "pypi", + "version": "==0.14.1" + }, + "wheel": { + "hashes": [ + "sha256:9515fe0a94e823fd90b08d22de45d7bde57c90edce705b22f5e1ecf7e1b653c8", + "sha256:e721e53864f084f956f40f96124a74da0631ac13fbbd1ba99e8e2b5e9cafdf64" + ], + "version": "==0.30.0" + } + }, + "develop": {} +} diff --git a/SETUP.md b/SETUP.md new file mode 100644 index 0000000..adeecb2 --- /dev/null +++ b/SETUP.md @@ -0,0 +1,7 @@ +#### Process +1. `$ ./setup.sh` downloads the avenue dataset to *./share/data/videos/avenue* +2. `$ pipenv install` installs the dependencies +3. `$ pipenv shell` activates the virtualenv +4. `$ python convert_video_to_frame.py` to start preprocessing the Dataset +5. `$ python start_train.py` to train the model. +6. `$ python start_test.py` to text the trained model. diff --git a/classifier.py b/classifier.py index eb601aa..7bc7d3f 100644 --- a/classifier.py +++ b/classifier.py @@ -55,7 +55,7 @@ def compile_model(model, loss, optimizer): model.compile(loss=loss, optimizer=opt) -def train(dataset, job_folder, logger, video_root_path='/share/data/videos'): +def train(dataset, job_folder, logger, video_root_path='./share/data/videos'): """Build and train the model """ import yaml @@ -76,7 +76,7 @@ def train(dataset, job_folder, logger, video_root_path='/share/data/videos'): time_length = cfg['time_length'] # shuffle = cfg['shuffle'] - logger.info("Building model of type {} and activation {}".format(model_type, activation)) + #logger.info("Building model of type {} and activation {}".format(model_type, activation)) model = get_model(time_length) logger.info("Compiling model with {} and {} optimizer".format(loss, optimizer)) compile_model(model, loss, optimizer) @@ -198,7 +198,7 @@ def calc_auc_overall(logger, dataset, n_vid, save_path): def test(logger, dataset, t, job_uuid, epoch, val_loss, visualize_score=True, visualize_frame=False, - video_root_path='/share/data/videos'): + video_root_path='./share/data/videos'): import numpy as np from keras.models import load_model import os @@ -306,4 +306,3 @@ def test(logger, dataset, t, job_uuid, epoch, val_loss, visualize_score=True, vi logger.info("{}: Calculating overall metrics".format(dataset)) auc_overall, eer_overall = calc_auc_overall(logger, dataset, n_videos[dataset], save_path) - diff --git a/convert_video_to_frame.py b/convert_video_to_frame.py index 9df60cb..b3c9207 100644 --- a/convert_video_to_frame.py +++ b/convert_video_to_frame.py @@ -3,7 +3,7 @@ import skvideo.io from skimage.transform import resize from skimage.io import imsave -video_root_path = '/share/data/videos' +video_root_path = './share/data/videos' size = (224, 224) def video_to_frame(dataset, train_or_test): @@ -28,17 +28,17 @@ video_to_frame('avenue', 'training') video_to_frame('avenue', 'testing') # ped1 -video_to_frame('ped1', 'training') -video_to_frame('ped1', 'testing') - -# ped2 -video_to_frame('ped2', 'training') -video_to_frame('ped2', 'testing') - -# enter -video_to_frame('enter', 'training') -video_to_frame('enter', 'testing') - -# exit -video_to_frame('exit', 'training') -video_to_frame('exit', 'testing') +# video_to_frame('ped1', 'training') +# video_to_frame('ped1', 'testing') +# +# # ped2 +# video_to_frame('ped2', 'training') +# video_to_frame('ped2', 'testing') +# +# # enter +# video_to_frame('enter', 'training') +# video_to_frame('enter', 'testing') +# +# # exit +# video_to_frame('exit', 'training') +# video_to_frame('exit', 'testing') diff --git a/dataset.py b/dataset.py index 61cc307..dd17307 100644 --- a/dataset.py +++ b/dataset.py @@ -1,5 +1,5 @@ -def calc_mean(dataset, video_root_path='/share/data/videos'): +def calc_mean(dataset, video_root_path='./share/data/videos'): import os from skimage.io import imread import numpy as np @@ -22,7 +22,7 @@ def calc_mean(dataset, video_root_path='/share/data/videos'): np.save(os.path.join(video_root_path, dataset, 'mean_frame_224.npy'), frame_mean) -def subtract_mean(dataset, video_root_path='/share/data/videos'): +def subtract_mean(dataset, video_root_path='./share/data/videos'): import os from skimage.io import imread import numpy as np @@ -56,7 +56,7 @@ def subtract_mean(dataset, video_root_path='/share/data/videos'): np.save(os.path.join(video_root_path, dataset, 'testing_frames_{}.npy'.format(frame_folder)), testing_frames_vid) -def build_h5(dataset, train_or_test, t, video_root_path='/share/data/videos'): +def build_h5(dataset, train_or_test, t, video_root_path='./share/data/videos'): import h5py from tqdm import tqdm import os @@ -87,7 +87,7 @@ def build_h5(dataset, train_or_test, t, video_root_path='/share/data/videos'): build_volume(train_or_test, num_videos, time_length=t) -def combine_dataset(dataset, t, video_root_path='/share/data/videos'): +def combine_dataset(dataset, t, video_root_path='./share/data/videos'): import h5py import os from tqdm import tqdm @@ -121,7 +121,7 @@ def combine_dataset(dataset, t, video_root_path='/share/data/videos'): output_file.close() -def preprocess_data(logger, dataset, t, video_root_path='/share/data/videos'): +def preprocess_data(logger, dataset, t, video_root_path='./share/data/videos'): import os # Step 1: Calculate the mean frame of all training frames @@ -134,6 +134,7 @@ def preprocess_data(logger, dataset, t, video_root_path='/share/data/videos'): testing_frame_path = os.path.join(video_root_path, dataset, 'testing_frames') if not os.path.isfile(mean_frame_file): # The frames must have already been extracted from training and testing videos + # import pdb; pdb.set_trace() assert(os.path.isdir(training_frame_path)) assert(os.path.isdir(testing_frame_path)) logger.info("Step 1/4: Calculating mean frame for {}".format(dataset)) @@ -182,4 +183,3 @@ def preprocess_data(logger, dataset, t, video_root_path='/share/data/videos'): combine_dataset(dataset, t, video_root_path) logger.info("Preprocessing is completed") - diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..6b8bc87 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,34 @@ +absl-py==0.1.11 +astor==0.6.2 +bleach==1.5.0 +coloredlogs==9.0 +cycler==0.10.0 +decorator==4.2.1 +gast==0.2.0 +grpcio==1.10.0 +h5py==2.7.1 +html5lib==0.9999999 +humanfriendly==4.8 +Keras==2.1.5 +kiwisolver==1.0.1 +Markdown==2.6.11 +matplotlib==2.2.0 +networkx==2.1 +numpy==1.14.1 +Pillow==5.0.0 +protobuf==3.5.2 +pyparsing==2.2.0 +python-dateutil==2.6.1 +pytz==2018.3 +PyWavelets==0.5.2 +PyYAML==3.12 +scikit-image==0.13.1 +scikit-learn==0.19.1 +scipy==1.0.0 +six==1.11.0 +sk-video==1.1.10 +tensorboard==1.6.0 +tensorflow==1.6.0 +termcolor==1.1.0 +tqdm==4.19.6 +Werkzeug==0.14.1 diff --git a/setup.sh b/setup.sh new file mode 100755 index 0000000..439fa86 --- /dev/null +++ b/setup.sh @@ -0,0 +1,6 @@ +#!/bin/sh + +wget http://www.cse.cuhk.edu.hk/leojia/projects/detectabnormal/Avenue_Dataset.zip +unzip Avenue_Dataset.zip +mkdir -p share/data/videos +mv Avenue\ Dataset share/data/videos/avenue diff --git a/start_test.py b/start_test.py index 37c3e06..cc2638d 100644 --- a/start_test.py +++ b/start_test.py @@ -13,7 +13,7 @@ epoch = 586 val_loss = 0.001069 time_length = 8 -job_folder = os.path.join('/share/clean/{}/jobs'.format(dataset), job_uuid) +job_folder = os.path.join('./share/clean/{}/jobs'.format(dataset), job_uuid) log_path = os.path.join(job_folder, 'logs') os.makedirs(log_path, exist_ok=True) logging.basicConfig(filename=os.path.join(log_path, "test-{}.log".format(datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))), diff --git a/start_train.py b/start_train.py index 08be7b7..2f2b247 100644 --- a/start_train.py +++ b/start_train.py @@ -7,11 +7,12 @@ from classifier import train import uuid from shutil import copyfile -dataset = 'ped1' +# dataset = 'ped1' +dataset = 'avenue' device = 'gpu1' job_uuid = str(uuid.uuid4()) -job_folder = os.path.join('/share/clean/{}/jobs'.format(dataset), job_uuid) +job_folder = os.path.join('./share/clean/{}/jobs'.format(dataset), job_uuid) os.makedirs(job_folder) copyfile('config.yml', os.path.join(job_folder, 'config.yml'))