Commit graph

257 commits

Author SHA1 Message Date
Laurent Fainsin dc4a399c0f feat: export to onnx
Former-commit-id: fd7e5a5ab785263a16381545ca31fd9e7fe86743 [formerly 10fdf9732fbcf4d922d945adc625e948e5f6e775]
Former-commit-id: 871745033b59e626fc38b38bfc8685c6a6366ecf
2022-06-30 11:44:20 +02:00
Laurent Fainsin 81938b944e feat: create overlay instead of table
Former-commit-id: e6a0ea3ce845ab6ff524f98cebd49151da5765de [formerly 1fc368fdaba7b892357848fd47a8e53ed34a2c35]
Former-commit-id: afc34abfffe0e309d6e4ae77d4d56d4f372b9e9d
2022-06-30 10:48:05 +02:00
Laurent Fainsin dac6237906 feat: reduce the number of parameters in the net
Former-commit-id: 862569b6d284ec8235586b161d8c7055c006f5d8 [formerly f2e672d780df12a398e851f375a238c2d394a3cd]
Former-commit-id: 740b1129a627c488537bb0d0dc7ff73b66fde813
2022-06-30 10:47:53 +02:00
Laurent Fainsin 9fe76d8c61 feat: prediction script
Former-commit-id: dcaba9f9fbeaec393cea168a16287c690c5733b0 [formerly c69d87581930858ca293326002588a0188431fe7]
Former-commit-id: 39fd4965182a2c4ae8ffac2f20c7dbaf4b82a61f
2022-06-29 16:12:00 +02:00
Laurent Fainsin c73b803a15 feat: randomize paste brightness
Former-commit-id: f3a14bc9fe5ea633837b638cc04313566e951190 [formerly 830199ee956eed9f62f845f1406d00c14e1f1262]
Former-commit-id: 542421d35c7103812cf621af5777c88812506adb
2022-06-29 15:01:57 +02:00
Laurent Fainsin e4b155991b feat: saving models with wandb
Former-commit-id: 994d061d49c8be9f680878d093b1e2b5710fa37c [formerly 3cadb0699c150421244c78511ee61b14262728d0]
Former-commit-id: 64b01c6ed73ba64a25d85acd65adc9b6c9cf126e
2022-06-29 14:15:04 +02:00
Laurent Fainsin c700835065 fix: paste bad algo :(
Former-commit-id: 23df9b5127d0abd3cc9ac77f56b0a1755b28eb00 [formerly 30326f6dd6eace91f4e0e463eab3fdb2083fb38e]
Former-commit-id: 23cbd859a96b5cf0cc361b71b32b54f54edb321b
2022-06-29 11:22:23 +02:00
Laurent Fainsin 83f8825340 fix: typo
Former-commit-id: 837b298afc0a8283bd18d0d4353c8f0491113b9a [formerly d64ed0d75f62f99b0be0a919d6d6bfffb6e5cd6f]
Former-commit-id: 964fb5b2dde3e90265dba16aef747b34dcfb6fb6
2022-06-29 10:26:26 +02:00
Laurent Fainsin de7e2e895f f5
Former-commit-id: c3c20fee02acb1c9eec7332ec173e05e9df9fb99 [formerly a0bdb02a4ec4299e767d011e86970412be10d25c]
Former-commit-id: 495e626a762ffa4d8df3f66dd6e3d7c896950f94
2022-06-29 10:20:35 +02:00
Laurent Fainsin 826ae4ac9b chore: ran the comp notebook
Former-commit-id: 8f033562a644885c8929ad89c2ea87ad559d07bc [formerly c35693f34e04d480c878a842061151df3e9ada70]
Former-commit-id: 923a2a2f32e1f88a230902620cdd01d1bf4210ad
2022-06-29 10:16:49 +02:00
Laurent Fainsin e089bcc7c4 feat: check for overlapping when pasting
Former-commit-id: 920a2df1e54e3dfc22cf62fe9c892bf945af9b63 [formerly f0c2daaa4d0094fafeddce9c07c6ce020c1704d6]
Former-commit-id: 7e74f04ed0cd0d31612f23a44b9832b659bf5408
2022-06-29 10:16:26 +02:00
Laurent Fainsin 9b624da832 chore: delete unused file
Former-commit-id: 540acfba838755064facc7a39bfe72b6cc0a2796 [formerly 085dd013a8e439d44b173a563b0fa8cd40cd9b75]
Former-commit-id: c6c8fffc2768ec54ba1adb1648de17415f130db7
2022-06-29 10:11:27 +02:00
Laurent Fainsin 806e98c744 feat: add some comments to evaluate.py
Former-commit-id: 0181b044ec0497d95c80e9e7c5112537806e0afc [formerly 1f62452a8fa1bbf6cfa5c82347c58cc1886ae796]
Former-commit-id: 31dba554cbd0126009890ac54dfd6b6d38f2a47c
2022-06-29 10:10:56 +02:00
Your Name 92ac3a2ab8 feat: better pasting function
Former-commit-id: 43fedd3f6bafb51fe604e347f59b70cd5b0cc218 [formerly 51bb06c3b98df613710b329d3ade1febaf2b0b23]
Former-commit-id: 46b89acd2b860d272ce8a13cf2c8c955d7545c46
2022-06-28 16:36:50 +02:00
Your Name 1e388c6b90 fix: ignore the INTERRUPTED.pth checkpoint file
Former-commit-id: 5097bb2efad0380e8847397a615de6e13bf5259f
2022-06-28 15:03:28 +02:00
Your Name 842aaa81c2 wtf am i doing
Former-commit-id: dde43cce52408ec8f67372b365796b9014ceee57
2022-06-28 11:36:43 +02:00
Your Name ad179241ce fix: broken Ups
Former-commit-id: 9c33326beb0d44e8491b040e25fad57ffa820076
2022-06-28 09:36:21 +02:00
Your Name d1083513f9 refactor: renamed a file
Former-commit-id: b1e3f616bbadd8087ad52b13152d4dc72d1267aa
2022-06-27 16:41:03 +02:00
Your Name 21c388ca2e style: formatting
Former-commit-id: 2ccef30ce44d33beb611b63adef635ab2c1226bb
2022-06-27 16:40:04 +02:00
Your Name a42190ec61 feat: made model.py extensible
Former-commit-id: 0261a3f6caf571dc19e6ca97d2ecada1e72c7f04
2022-06-27 16:31:07 +02:00
Your Name bc2aeacfa3 refactor: removed bilinear stuff + simplified the construction of the Downs and Ups
Former-commit-id: 4c1e0a5a9fc02047b788b13d9bfc3ad7313413e3
2022-06-27 16:13:38 +02:00
Your Name cc4e8089ec refactor: renamed files
Former-commit-id: c9936db9c051268c7e9bd3a78bbeaab9948e416e
2022-06-27 15:47:21 +02:00
Your Name 30b9abfc5b refactor: moved code to src folder
Former-commit-id: 5ec2749618d346e8adc40da1e2931a4e02ce8f41
2022-06-27 15:42:03 +02:00
Your Name a71e67690a style: autoformating
Former-commit-id: 8c5c75469afa61e8d3728959390b1354033be462
2022-06-27 15:39:44 +02:00
Your Name cae795b9ad chore: cleaned the repo
Former-commit-id: 7507ab86f58012d749e0752ea2cb05937d3d8a32
2022-06-27 15:28:28 +02:00
milesial 408f2c9ec2 Merge pull request #353 from Gouvernathor/patch-1
Various minor changes

Former-commit-id: e36c782fbfc976b7326182a47dd7213bd3360a7e
2022-04-11 23:56:51 +02:00
milesial e9d50a9524 Merge pull request #356 from qslia/qslia
print assert error correctly

Former-commit-id: f0170cc7c0c2b97709ef4b0249cbb1c55b29e15f
2022-04-11 23:55:26 +02:00
qslia 109cf3462b change args name
Former-commit-id: 22020bdda52ed24e2163083dc1fa6889ba266225
2022-04-10 11:41:35 +08:00
qslia 52ced2e757 print assert error correctly
Former-commit-id: 8c7feff7cf740e788228839883a592c87097add2
2022-04-10 07:54:48 +08:00
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