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
milesial
ef36cad9ba
Add note to use python 3 to README
...
Former-commit-id: fd16a756231b24b5d51810244415a70198492ef9
2018-12-28 17:56:15 +01:00
milesial
d3150e8923
Fix upsampling issues (maybe) ( #38 , #32 )
...
Former-commit-id: 0f45521480d2de8a93407ce43a2a813eefd5d1f6
2018-11-19 11:51:46 +01:00
milesial
5da5e17c08
Merge pull request #37 from rht/master
...
Move the sigmoid activation to the model itself
Former-commit-id: 7dd7c8b6346033ed1b15a2c1ab54847016a826db
2018-11-11 20:56:15 +01:00
rht
81234be7da
Move the sigmoid activation to the model itself
...
Former-commit-id: e3f8ca7b1ac7c5e9694637a81be260e9b48973b9
2018-11-10 23:11:52 +00:00
milesial
46d1db3115
Another error in diceloss
...
Former-commit-id: 24a26eaa42b2a8e24335a421047af37cd11adb85
2018-10-28 14:54:14 +01:00
milesial
7e5a8727de
Fixed math error in DiceLoss backward pass ( #35 )
...
Former-commit-id: 649124b3bd26272bb8442843a546c4e895ec14ff
2018-10-26 18:53:05 +02:00
milesial
f2d88f98a7
Switch net to eval mode in predict.py
...
Former-commit-id: 3af421f3060c30b19ee38bc6e00db2593e53ad62
2018-09-26 08:59:49 +02:00
milesial
341824fb06
Switch net to train mode in train.py
...
Former-commit-id: ab255c4b3b823dfe8681083aeac0ee78c1a54780
2018-09-26 08:58:49 +02:00
milesial
b1ac7151b4
Switch net to eval mode in eval.py
...
Former-commit-id: 3ab827241455818c7b81172a426d805f9ace0f8f
2018-09-26 08:57:10 +02:00
milesial
3fefc25199
Fixed misuse of epsilon in Dice loss (issue #24 )
...
Former-commit-id: 1e8093de0398364cc65f3c7722b61a0f560df104
2018-08-16 19:50:24 +02:00
milesial
f991ffa23e
Merge pull request #20 from changjo/patch-1
...
Fixed argument actions
Former-commit-id: 23e817310461f79888078a2a1ab4b2187a97dc88
2018-07-10 13:17:51 +02:00
Chang Jo Kim
179b08e6a3
Fixed argument actions
...
Former-commit-id: 7f855d53668a576236b16670fd052d8caf816d80
2018-07-10 11:19:03 +09:00
milesial
14e0ff1c62
Merge pull request #13 from ZiyuanTonyZhang/master
...
fix scaling issue
Former-commit-id: b282e761f04904697418c9e759d761004a045428
2018-06-23 19:59:38 +02: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
af08dbb98d
Created a requirements.txt
...
Former-commit-id: ec6b0a4fc245bea8e073fe942cd519e31f1b764f
2018-06-08 19:28:46 +02:00
milesial
02e2314149
Migration to PyTorch 0.4, code cleanup
...
Former-commit-id: c981801ccc3b74047e94c76e67c4ff1f3097226c
2018-06-08 19:27:32 +02:00
milesial
90e988c10f
Fixed bad use of ConvTranspose2D
...
Former-commit-id: 245b072a82f2e87a9032a1b87136a9930c008afc
2018-06-07 11:20:35 +03:00
