Commit graph

278 commits

Author SHA1 Message Date
Gouvernathor 99e36df431 change the order of asserts
Former-commit-id: 6cd4b68c997542cf0a9ca205120fd4f275e3cb24
2022-04-06 13:45:19 +02:00
Gouvernathor 8848edf8f3 add wandb
Former-commit-id: a3a9fd1ab94b3d24a991a532feb64b0df12ccddf
2022-04-06 13:44:15 +02:00
Gouvernathor 758f0f8070 Update train.py
Former-commit-id: dee78b12ca6810f5e02febfb244dce3885ed49ac
2022-04-06 13:35:02 +02:00
milesial 182cbd48f9 Merge pull request #347 from Gouvernathor/patch-1
Simplifying some code

Former-commit-id: 68d59a94dae1c5eeeb79c689d0d7206e320fa0c9
2022-03-31 21:40:17 +02:00
milesial c1c7c952aa Merge pull request #348 from pleaseRedo/master
Update utils.py. minor fix

Former-commit-id: 1c736851c43aceb70f49f6e444f1e22a377ec59d
2022-03-31 21:37:48 +02:00
pleaseRedo eb95dab01e Update utils.py
fix visualization glitch, when running predict.py


Former-commit-id: 291e41e01bbcd783c3b7ac4b9e59d9d8427e0bf5
2022-04-01 01:07:13 +08:00
Gouvernathor 7f800aff75 Simplifying
- avoid using classmethod if the class argument is not used, use staticmethod instead
- avoid testing for is_mask three times if one is enough

Former-commit-id: 1116e3096cd5210a0596d1f3938989016e0f3b6c
2022-03-31 16:23:37 +02:00
milesial 2ca43802cc Fix torchhub
Former-commit-id: 9a6ee84c50390306e24bb369d1c24d2fdd9d7e60
2022-02-19 05:15:45 +01:00
milesial f55693b6a4 Update PyTorch container, fix data download
Former-commit-id: 7679965b6814e40a9bc7be694d30cfd4bb29b9f7
2022-02-19 02:38:49 +01:00
milesial c1f4757c32 Merge pull request #325 from likyoo/master
delete convolution bias before BN

Former-commit-id: 0921374ee4ed02688ef5ed48c50b241ee8911069
2021-12-13 12:02:03 +01:00
Kaiyu Li a953415a0d delete convolution bias before BN
The convolution bias before BN is redundan

Former-commit-id: 514f18e1a961fee4bf98fa6f50084a166e3bff4b
2021-12-13 14:51:44 +08:00
milesial 473a03ce17 Merge pull request #318 from Liunaijiaaa/master
Fix wandb error in multi-category segmentation

Former-commit-id: e1a69e7c6ce18edd47271b01e4aabc03b436753d
2021-11-13 10:28:59 +01:00
milesial 02fc4ff733 Quick start without docker
Former-commit-id: c4555b6d1ae960f8d253b1e45c25e2279b64ead5
2021-11-13 10:07:03 +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
milesial a445c4bdc4 Fix mask resampling in some cases
Former-commit-id: 8f317cb13c17ef25a86b25a0c24390e04cd4db82
2021-11-03 09:39:21 +01:00
milesial 229dd44fca Merge pull request #308 from Arka161/bug_fixes
Fixed bugs for no validation set, and potential module zero error

Former-commit-id: 4f283f1b7e47992a2af1bc738c54562d5cd117c8
2021-10-25 15:05:03 +02: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 c912d9b726 Switch to reshape in dice coeff
Former-commit-id: db72295019a2114f4c84940d9aaf1232b2a23352
2021-08-31 17:54:02 +02:00
milesial 449965b398 Fix evaluation on 1 class
Former-commit-id: d8984977848d924644d575f5e46c475d84ff1772
2021-08-21 10:26:42 +02:00
milesial 8c761f9d06 Fix evaluation
Former-commit-id: 2b649bc9a337818696291280cd87ad93a6fc5032
2021-08-21 08:41:23 +02:00
milesial 890719f4b7 Update README.md
Former-commit-id: 5a7a0d5a57467e43e8d10c9b28a99b67e0936803
2021-08-19 11:15:41 +02:00
milesial 1f1967d271 Update README
Former-commit-id: 8ebf8f874be976cace2d32365c8dce31fe7d6d5a
2021-08-19 11:14:36 +02:00
milesial 1c450fa978 Update torchhub
Former-commit-id: 7a89332de99777c24419d8ed4154bb1963ddc8f5
2021-08-19 11:14:08 +02:00
milesial fdb686fc43 Increase shared memory
Former-commit-id: 3a5f3a0c8c5069e837ade1997df7139f482bc985
2021-08-19 10:42:39 +02:00
milesial ce4567c908 Fix import in utils
Former-commit-id: faa6e13ea9b53d6773e465a2a60617ffc3d27939
2021-08-19 09:40:54 +02:00
milesial 4e638b47de Update README
Former-commit-id: 20d8d2b9446970ac74deeb7c36a65ae9e8819a29
2021-08-19 00:30:35 +02:00
milesial c1dcc0877e Remove license badge
Former-commit-id: b62770ef739c10ae96e787197979bbfd3a0dbc7c
2021-08-18 00:57:41 +02:00
milesial 581c423f62 Add CI badge
Former-commit-id: 1f5b4b71ca573469fad2754186ff810673a1fa3a
2021-08-18 00:56:56 +02:00
milesial 4b819f4df8 Fix actions
Former-commit-id: 016536a695baf8ea78b83da6b4b2bfe121e070e9
2021-08-18 00:31:19 +02:00
milesial f0381301bb Fix actions
Former-commit-id: e9e1884fb7728e622357ec0ce1705fa2c845248c
2021-08-18 00:26:11 +02:00
milesial 6f3320edbc Add action
Former-commit-id: 30c4e6a69c339ed456c912f56af6a05f8090d357
2021-08-18 00:19:59 +02:00
milesial cb238c82c5 Update README
Former-commit-id: 227a64375cf3a078aa9d3163af76f67b4d3dfdc4
2021-08-17 23:09:50 +02:00
milesial 6c057cc16a Add badges
Former-commit-id: f7441d0dac9d266525bc3c7d17e668535e65d03f
2021-08-17 22:59:34 +02:00
milesial 218fc38ca5 Fix README instructions
Former-commit-id: 3675576533ceb82671f2ee01eca0adb55786dd8d
2021-08-17 22:25:24 +02:00
milesial 27e87e14c9 Add ToC to README
Former-commit-id: 7ff11399bda293718b868aef79d8b438e9a18c2f
2021-08-17 17:18:19 +02:00
milesial 1022338a3b Add docker instructions
Former-commit-id: b44a96b989ff3000a5afa72af863f4cb0df21e92
2021-08-16 16:54:06 +02:00
milesial 9bd46dab50 Fix README link
Former-commit-id: 6cc58d6930ae97d31665709eaedbabb4c4edb804
2021-08-16 07:22:02 +02:00
milesial cb0c274489 Update default number of epochs
Former-commit-id: 450752536d8c72ad978e0ac319c624f537444d96
2021-08-16 06:46:52 +02:00
milesial 423e94868a Update README
Former-commit-id: 8bdc2a9746e1745de3856267ae27460c04b8d07e
2021-08-16 06:21:40 +02:00
milesial 5cd2a3b0b7 Restructure
Former-commit-id: beadb49b75ea79a3c0f95df589f64a8274419c5b
2021-08-16 06:08:09 +02:00
milesial a7270ca699 README update
Former-commit-id: c551c91786f26acc93abed0c043115ab8ef2fce0
2021-08-16 06:04:25 +02:00
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