Commit graph

9 commits

Author SHA1 Message Date
milesial 063cbbc599 Summer cleanup
Former-commit-id: f6185d67a4bc50aa7ec1b8168aab3f92721c4965
2021-08-16 02:53:00 +02: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 490ecf8383 Rework of the transposed conv / bilinear up route
Former-commit-id: 07debeb9f2621b53ed513e5ab8a0307b4da57767
2020-03-30 20:17:52 -07: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 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 02e2314149 Migration to PyTorch 0.4, code cleanup
Former-commit-id: c981801ccc3b74047e94c76e67c4ff1f3097226c
2018-06-08 19:27:32 +02:00
milesial 7ea54febec Added CLI for predict, cleaned up code, updated README
Former-commit-id: 77555ccc0925a8fba796ce7e42843d95b6e9dce0
2017-11-30 06:45:19 +01:00
Renamed from unet_model.py (Browse further)