Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-09-19 19:45:28 +00:00
Code Issues Packages Projects Releases Wiki Activity
132 commits 1 branch 0 tags 38 MiB
Commit graph

32 commits

Author SHA1 Message Date
Gouvernathor 758f0f8070 Update train.py 
Former-commit-id: dee78b12ca6810f5e02febfb244dce3885ed49ac
 2022-04-06 13:35:02 +02:00
milesial 2ca43802cc Fix torchhub 
Former-commit-id: 9a6ee84c50390306e24bb369d1c24d2fdd9d7e60
 2022-02-19 05:15:45 +01:00
Liunaijiaaa 06199357af Solved the problem that wandb could not display masks_pred in the multi-category segmentation problem.If we do not modify it, we may encounter the following error message:mode, rawmode = _fromarray_typemap[typekey] KeyError: ((1, 1, 15), '|u1').ps:15 is the number of categories. 
Former-commit-id: ff0c2ffa33c945a491e85ef44e89891e9a7c435d
 2021-11-13 16:55:50 +08:00
Arka 2525a30b9d typo 
Former-commit-id: 9b99dba1e6ad5fec942f75418676330f21cef7c2
 2021-10-24 17:14:18 -04:00
Arka 6438b1dcdd bug fixes for i/o and low val set 
Former-commit-id: bdcad8300e3c930b43b976ccd2562f27c9867892
 2021-10-24 17:07:54 -04:00
milesial 1022338a3b Add docker instructions 
Former-commit-id: b44a96b989ff3000a5afa72af863f4cb0df21e92
 2021-08-16 16:54:06 +02:00
milesial cb0c274489 Update default number of epochs 
Former-commit-id: 450752536d8c72ad978e0ac319c624f537444d96
 2021-08-16 06:46:52 +02:00
milesial 5cd2a3b0b7 Restructure 
Former-commit-id: beadb49b75ea79a3c0f95df589f64a8274419c5b
 2021-08-16 06:08:09 +02:00
milesial 063cbbc599 Summer cleanup 
Former-commit-id: f6185d67a4bc50aa7ec1b8168aab3f92721c4965
 2021-08-16 02:53:00 +02:00
milesial d43eaa1115 Pretrained model in the README 
Former-commit-id: 67bf11b4db4c5f2891bd7e8e7f58bcde8ee2d2db
 2020-08-12 00:42:01 -07:00
milesial 63f889a4b0 Fix eval frequency 
Former-commit-id: 1c6a9505c964fdd1bc7d55887725551afc94b956
 2020-06-11 20:53:00 -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 d40a71737d LR scheduler + TensorBoard weights and grad hists 
Former-commit-id: 0e7a96225ecc7761419aa67fef861ff6afae4c55
 2020-03-15 21:38:51 -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 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 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 8ed1e09b2a Better support for multiclass 
Former-commit-id: 76bebf5f241f579fda7048f5e4a87ee9d49aa423
 2019-10-30 19:54:57 +01:00
milesial 4c0f0a7a7b Global cleanup, better logging and CLI 
Former-commit-id: ff1ac0936c118d129bc8a8014958948d3b3883be
 2019-10-26 23:17:48 +02:00
rht 81234be7da Move the sigmoid activation to the model itself 
Former-commit-id: e3f8ca7b1ac7c5e9694637a81be260e9b48973b9
 2018-11-10 23:11:52 +00:00
milesial 341824fb06 Switch net to train mode in train.py 
Former-commit-id: ab255c4b3b823dfe8681083aeac0ee78c1a54780
 2018-09-26 08:58:49 +02: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 7ea54febec Added CLI for predict, cleaned up code, updated README 
Former-commit-id: 77555ccc0925a8fba796ce7e42843d95b6e9dce0
 2017-11-30 06:45:19 +01:00
milesial 932e1c3cf5 Added simple trained model 
Former-commit-id: e87526c043c447e5c8d95ee8314fcb7f97c13ce2
 2017-08-23 17:38:55 +02:00
milesial fa40396fff Added simple eval and test CRF 2017-08-19 10:59:51 +02:00
milesial 4063565295 Created a basic train loop + changed a bit loss and utils 2017-08-17 21:16:19 +02:00
milesial c8c82204bf First prototype 
slow, no gpu, no validation
 2017-08-16 14:24:29 +02:00