Commit graph

186 commits

Author SHA1 Message Date
milesial c5257f8367 Data download script
Former-commit-id: 88ff94e36c5c8a3f2965c54ada19f39e539229cd
2021-08-16 03:18:09 +02:00
milesial 063cbbc599 Summer cleanup
Former-commit-id: f6185d67a4bc50aa7ec1b8168aab3f92721c4965
2021-08-16 02:53:00 +02:00
milesial 2f4f7edd5d Update image link in README
Former-commit-id: 6aa14cbbc445672d97190fec06d5568a0a004740
2021-01-21 14:43:08 +01:00
milesial 3b2a767694 Merge pull request #219 from ChaosPredictor/patch-1
Update requirements.txt (add tqdm)

Former-commit-id: 381924dfd6396c92b17ceebaa9c607f3f9392106
2020-09-30 13:27:12 +02:00
ChaosPredictor 74b5011f01 Update requirements.txt
Former-commit-id: 5e956a0fb95ed293dea7850409ef07bc400d563c
2020-09-29 20:55:58 +03:00
milesial d43eaa1115 Pretrained model in the README
Former-commit-id: 67bf11b4db4c5f2891bd7e8e7f58bcde8ee2d2db
2020-08-12 00:42:01 -07:00
milesial c3c2675369 Torch hub
Former-commit-id: 4ce26d9d02760fead8fa34445f0e01acf9e4efef
2020-07-29 18:50:33 -07:00
milesial 8780e424b4 Carvana dataset loader
Former-commit-id: 4ad8323b2e54c7bbcf8968ca7fc0f2e3ddd87689
2020-07-29 18:21:17 -07:00
milesial 04c2cb1ed4 Update README with xs:code
Former-commit-id: 84f8392b619940bd542dc670761a0a7a1357001d
2020-07-23 17:04:38 -07:00
milesial cc2ac3db07 Fix device in submit.py
Former-commit-id: 2eb8a6ff539ac0f4838ce1161b6d239e912ca007
2020-07-14 11:37:05 -07:00
milesial 63f889a4b0 Fix eval frequency
Former-commit-id: 1c6a9505c964fdd1bc7d55887725551afc94b956
2020-06-11 20:53:00 -07:00
milesial fb5f60fb31 Merge pull request #188 from laclouis5/batchSizeGreaterThanOneFix
Fixed Dataloader for batch size greater than one

Former-commit-id: 5419352796a0b221c2d8f9c47f2f7c1db2459c54
2020-06-05 16:35:12 -07:00
Louis Lac 416e076dbc Conversion to FloatTensor
Former-commit-id: cc44eebf19f5d98fbd1ca800608ae0b38d998b54
2020-06-05 19:00:04 +02:00
milesial aa9defe6a5 Merge pull request #187 from laclouis5/development
Fixed glob in dataset

Former-commit-id: 15df8266027d48ec1e78610338283c2a5dca7f7b
2020-06-03 14:24:32 -07:00
Louis Lac 4db3cb4620 Fixed glob in dataset
Former-commit-id: 88a44f9fdd08df1e0d87efb706506dbd0898f6fd
2020-06-03 15:47:19 +02:00
milesial 9fb406b393 Clarification in README
Former-commit-id: 02644ec8e673f54006c8cc1523610e149e84ceb5
2020-06-01 11:20:34 -07:00
milesial 0731ea2ff2 Update python version in README
Former-commit-id: e5039519d57e39e7efb91183e29d39c049ba3100
2020-05-21 23:34:03 -07:00
milesial b83c81f30b Change back to train mode after eval
Former-commit-id: 773ef215d41c1f36dc0ed4159c12df89d792fbc3
2020-05-05 09:15:07 -07:00
milesial 74a3f2496f Merge pull request #167 from HansBambel/master
diffX and diffY is now an int instead of a Tensor

Former-commit-id: d66f2c3d35c1c7dc58c6d35c6ee6aa05cacfbccf
2020-04-24 10:59:11 -07:00
HansBambel ef14c0e829 diffX and diffY is now an int instead of a Tensor
Former-commit-id: 342d4e0a8ca19bbee77256caeda0ef6537d746f8
2020-04-24 15:57:53 +02:00
milesial c57180e90c Merge pull request #161 from somehower/master
optimize the passed in parameters in the case of bilinear

Former-commit-id: 5f37e8a6dc592563593210371cebd40f95788080
2020-04-17 09:29:20 -07:00
hushenghao b5a3b207a1 optimize the passed in parameters in the case of bilinear
Former-commit-id: 6234bd1c4608cac8a64d8658e9af82b60a227859
2020-04-17 23:00:11 +08:00
milesial 8b1fee8730 Merge pull request #151 from joshmyersdean/master
Fix tensorboard dependency issue

Former-commit-id: 7afb4b25ba5be0da8177ff3f49a8262a3e5f7498
2020-04-01 11:57:23 -07:00
joshmyersdean b1b7d9346b Fix tensorboard dependency issue
Former-commit-id: 9aa2e01e5169a685ecccad780b144cc28c23e919
2020-03-31 20:25:38 -07:00
milesial 490ecf8383 Rework of the transposed conv / bilinear up route
Former-commit-id: 07debeb9f2621b53ed513e5ab8a0307b4da57767
2020-03-30 20:17:52 -07:00
milesial 2bef826dcc Add tensorboard.dev link in the README
Former-commit-id: d685fe10ac571e28e818a9d8fb19013e3346ee5f
2020-03-15 22:37:20 -07:00
milesial d40a71737d LR scheduler + TensorBoard weights and grad hists
Former-commit-id: 0e7a96225ecc7761419aa67fef861ff6afae4c55
2020-03-15 21:38:51 -07:00
milesial 54ba0e5d54 Merge pull request #141 from whenyd/master
Apply sigmoid before calc dice in eval_net()

Former-commit-id: 32ee0fe21170e98f3edb51f8639d8b26d7ce3475
2020-03-14 15:02:50 -07:00
whenyd cf505c6549 Set drop_last=True for val_loader
Former-commit-id: adb1d7e9348a2707f38d9b86f57bb3ae2cbc2b73
2020-03-14 13:19:00 +08:00
whenyd d292e8c6cd Apply sigmoid before calc dice in eval_net()
Former-commit-id: 0da18fda34f29c81968425715e19c5dc76c9ec46
2020-03-14 13:16:52 +08:00
milesial d081192e90 Merge pull request #109 from ant-Korn/save_to_onnx
Corrected class Up for model to ONNX exporting

Former-commit-id: 060bdcd69886a3082a6f8fb7746e12d5fca3e360
2019-12-29 20:02:03 +01:00
Anton Kornilov 70e6ac4258 Corrected class Up for model to ONNX exporting
Former-commit-id: 2dceff8c909b16530e1291118a2165ea95744afd
2019-12-29 18:49:12 +03:00
milesial c0adecd5d6 Removed dense_crf (for real)
Former-commit-id: eb54ef62c6c893ab9e93d4a8095b75044c2d33a7
2019-12-27 18:30:23 +01:00
milesial 012fca4715 Removed dense_crf and small fixes
Former-commit-id: de7507ff08510b48e6a0e11da849e0d1c94d3ac8
2019-12-21 22:04:23 +01:00
milesial 5f4ce7dba9 Update mask type for muticlass
Former-commit-id: 4dcb7b8440c5f36ff2565c67f56f8f029b589c80
2019-12-13 17:36:12 +01:00
milesial 4e1f0398a1 Remove assert for multiclass
Former-commit-id: 16809ec7ba14f67bfa5e5a095d3b512ef669a964
2019-12-13 15:58:45 +01:00
milesial 67c4dd34a0 Use classmethod for preprocess
Former-commit-id: ebe3bbb3ca67502db407eaa8273071a871d4b744
2019-12-11 21:57:45 +01:00
milesial 0d1cc25ae2 Convert to tensor in predict.py
Former-commit-id: be1ad32304f2365c0c1db6b0e7fb835acc0fbfed
2019-12-04 12:43:04 +01:00
milesial 5a7e934560 Fix eval.py iterating within batch
Former-commit-id: d4b8040593d4bc82774b1dac647c701b2486c477
2019-12-02 12:06:29 +01:00
milesial 48f45274b7 Update README.md
Former-commit-id: 7cb24a105aa4261eb6daafbb9b0552fd0fdf6fd1
2019-11-23 18:09:00 +01:00
milesial 9d7be6e234 Cleanup + now using tensorboard
Former-commit-id: 79928c84cdf990ef6fe1043a3e4f74b9cc252642
2019-11-23 17:56:14 +01:00
milesial 35f955cbf8 Now using utils.data.Dataset
Former-commit-id: c75d9c075e18add5cd8683faf827937393bf2c94
2019-11-23 14:22:42 +01:00
milesial f5c2771242 Fix predict script (mistake from last commit)
Former-commit-id: 9fd386f5929945f3df1868930d392507fee16a55
2019-11-06 13:23:01 +01:00
milesial 8ed1e09b2a Better support for multiclass
Former-commit-id: 76bebf5f241f579fda7048f5e4a87ee9d49aa423
2019-10-30 19:54:57 +01:00
milesial 6f23624412 Fix typo
Former-commit-id: 72f12645410e87ba18093f055c9d8d9f8045595b
2019-10-30 12:27:03 +01:00
milesial 4c0f0a7a7b Global cleanup, better logging and CLI
Former-commit-id: ff1ac0936c118d129bc8a8014958948d3b3883be
2019-10-26 23:17:48 +02:00
milesial 74f825ce06 Fixed major data loading bug : #46 from pierrezeb/patch-1
Fix split_ids (returned only (id, 0))

The network is now training on both squares and not only the left part of the image

Former-commit-id: e2e46ce509382a45b1db4e1f639aeed568f6cb3e
2019-02-07 13:30:53 +01:00
pierrezeb 543e9e560c fix split_ids (returned only left square!!)
Former-commit-id: a357099e4eace375489f18babb2947d9103733fd
2019-02-07 16:43:47 +09:00
milesial 0f5e7cb0f9 Merge pull request #45 from dmitrysarov/master
evaluation fix : index + 1 in average

Former-commit-id: c8179b82ebe715817890a967b6775ef7875007b3
2019-01-09 14:32:55 +01:00
dmitrysarov 078b03ffa9 index + 1 in average
Former-commit-id: 1dc82b17f075a74f7ef6b20c888835497bcd690b
2019-01-09 15:01:42 +03:00