Commit graph

224 commits

Author SHA1 Message Date
Laurent Fainsin c471342681 chore: ran some notebooks
Former-commit-id: 8580d48d5783d1e34e34da8ece060eadeb6fb693 [formerly d6bedefa8920e3455bd757f053cf621e81901967]
Former-commit-id: 5485a8b1b93627e93794e7b80ae56bbaefccdea5
2022-08-26 11:10:58 +02:00
Laurent Fainsin 8d6115afb2 feat: add more config fields
Former-commit-id: 4b288d6c47aec460704348a003b48620327c218d [formerly 1af19e81906cd94483f7e9aadc7fdd72d35dc5e1]
Former-commit-id: 605c529e3db6dde13cbbcf41f9517c3380b69b29
2022-08-26 11:10:49 +02:00
Laurent Fainsin 382b3ff91a hotfix: save model on epoch end
Former-commit-id: 8cd802c53a27fff273746dd9c77305fd22f250ce [formerly 3172f1b49c82ed8d3405488983156ae3f62e0598]
Former-commit-id: cad3afc409deb9ffc905d1927383799850241e68
2022-08-26 11:10:07 +02:00
Laurent Fainsin 4ce22005cf feat: basic maskrcnn loss logging
Former-commit-id: 2180d68e979deffb743fecb6b34f5d1a20c7d729 [formerly bf1933b5b68e107d94c3f0834fb7321407c4bf1e]
Former-commit-id: 64219fc194f5f2ea41a4922afbc1e7d6dd9ee06a
2022-08-26 11:09:25 +02:00
Laurent Fainsin 4696885a30 feat: directly resize images to 1024px
Former-commit-id: b6ed46be0e93bc735adc5a5c912893e5978ab2c5 [formerly 18f3b64ee956824024be1e4b13075ebb12b86176]
Former-commit-id: dbeaa87c8900e6191cf322bbac8994c6d9939aca
2022-08-26 11:09:05 +02:00
Laurent Fainsin 313d491143 fix: wrong xy coordinates when pasting
Former-commit-id: 8d3f4982614fb9874aa26b5d682148b53d45231c [formerly 03f6799ff09de7314e1e420eb123de7a3523d8dd]
Former-commit-id: 9a8a5c2d21e49a72dfcc8b5efdcce3a20f4f8de6
2022-08-26 10:50:47 +02:00
Laurent Fainsin a876a741f0 fix: don't reuse the same image over and over when resizing
Former-commit-id: 182f0ed044dfebda406ee93a9a1071c7b37e2a2d [formerly 3f11f436feb764fe04c6b6b1ba41e2dcd4d6a986]
Former-commit-id: 81db41632a11963da271b24b1fd8b087aa2909af
2022-08-25 16:21:23 +02:00
Laurent Fainsin 5bd2e5b2c4 feat: it's training, but for how long ?
Former-commit-id: edc9e9bc0c32a08b263c945b9296980b5242924b [formerly 41b1c1a9704f82518e44dab52adc482e02cbbf73]
Former-commit-id: e157962777b0cf057628b6c127ee68537731a528
2022-08-25 10:40:50 +02:00
Laurent Fainsin 4dab157dda feat: WIP, replacing U-Net by Mask R-CNN
Former-commit-id: f51a572adac901ff588e3a467f39ecd26376e617 [formerly 376595d7e5f906928379e25c1246e304b96b156d]
Former-commit-id: 3f4772ba3483702be6e5f7a29f06be93eb1f3bb2
2022-08-24 14:56:41 +02:00
Laurent Fainsin c50235bb1e feat: changed DIR paths + weird optimization
Former-commit-id: c85c819a022a4cc9fffa88e833d1798b38d5a600 [formerly fb78d7be52badb87cc670bcb8bf0a83c5648d315]
Former-commit-id: 442152b8b84db736380cc17009f1bd4329bb6a22
2022-08-23 15:27:23 +02:00
Laurent Fainsin 0cb52febd7 chore: various files from 3 weeks ago
Former-commit-id: 502f9821bddd45574a46cdfe017524a6fac9f102 [formerly 5e2f3679b65b8ae554e1cdd3e1c13ae190e07bd0]
Former-commit-id: 6337e42ef9b051064a9cb2716ef37b272b09673e
2022-08-23 13:09:44 +02:00
Laurent Fainsin fb7e33c437 feat: threshold doesn't matter
Former-commit-id: a9f448ca7eb97b821065d158a2834b309be06443 [formerly d796c40cbabd787cf2726d4726075ab359a37d23]
Former-commit-id: 781ae46cdf3157cbe483edaca95cf8f6630ef501
2022-07-13 16:30:33 +02:00
Laurent Fainsin 8b46d0ce2d chore: ran the predict notebook
Former-commit-id: 31bd069590422a692caa84d50323ab05d960c568 [formerly 645498dd1cf4cdb0fe5e37f45cc5c22eeb0b55f7]
Former-commit-id: 248460b6fd4d4e26274839c614fa1edf5eca329e
2022-07-13 16:13:50 +02:00
Laurent Fainsin f431947693 feat: yay better post processing
Former-commit-id: a89f98871813209f1a9f68f15dfe5b59a59e94ca [formerly 4b79c0639edec6ca14444b1305141e88cff63950]
Former-commit-id: 921a54dba0d59391c286297092c8cf3a346c83ac
2022-07-13 15:10:27 +02:00
Laurent Fainsin 8fa2cde787 feat: yay more drafts for filtering output of unet
Former-commit-id: 8c46294df7a834e6b67043afbb905b5fcc8194f2 [formerly 79b7ecefd45e16c01afc3e062780147d14da127a]
Former-commit-id: e334164ea7c9b6c9093935d7e468efa66b4ccbe5
2022-07-12 16:17:37 +02:00
Laurent Fainsin dc833d2a88 feat: confiming synth "overfitting"
Former-commit-id: 9a6d691503c72e76ac68eab378fc07f0e35f5182 [formerly 9c0469156cf11b9d7540c52115f9fb20ce873d5d]
Former-commit-id: c597ef791d930739a3c56ba62e3b0070bff0e82b
2022-07-12 14:21:23 +02:00
Laurent Fainsin 50d18a5b39 feat: random code I don't want to make commit messages for
Former-commit-id: b94db28e25c4ada7f69d65185198a701bb5d6bfd [formerly 2476ee5d84287e40c8fb341f569249dc8aaff3e5]
Former-commit-id: 0a4b7a1f925165172b009f8812d3083e70f10201
2022-07-12 11:18:03 +02:00
Laurent Fainsin 9b7418facd feat: predict notebook
Former-commit-id: f0e1227a1374b5b994aa3fc80eee0dcbf9e4f04f [formerly 18575f832e0c39c7db0753a5eac7320134555c9b]
Former-commit-id: c35113f0eae831658e917705054633bfdf3507c8
2022-07-12 11:05:19 +02:00
Laurent Fainsin 82682ceeb2 feat: got precision 16 back
Former-commit-id: 6b19dc9bd17078bb2c151d5cd96e7ba4da9e1b89 [formerly 5d1eac2ed10be960c89407ad265ff350e11c1adf]
Former-commit-id: 1db4ca0ce11ac818408b94625b872c1202b5d4ed
2022-07-11 17:02:13 +02:00
Laurent Fainsin ed07e130e6 fix: small bugs
Former-commit-id: 61b601b392e0199527e13ab22658b81a87efaa37 [formerly c64178fb281cd142cb4cbbf9b9d4d63dfc0099e5]
Former-commit-id: 74f22879d2630647e9d853db8d1a21a889cab651
2022-07-11 16:13:16 +02:00
Laurent Fainsin 75a4907591 chore: moved notebooks in their own folder
Former-commit-id: 6fe8ce843274ad51209ca4ffcd4d490f6af565cf [formerly 2fdc849790bddc6230a57d66779d28c0662c2386]
Former-commit-id: c896d9d911be6c8ca007bf835c899bd5027d9830
2022-07-11 15:37:36 +02:00
Laurent Fainsin 47c888cf6c feat: checkpoint wandb logging
feat: wandb config file

Former-commit-id: 45f56db86ca269b028cf76bf5315bc0eef8d2a21 [formerly e320b72e16eed02bdca05245e7c77914f0e288f9]
Former-commit-id: 6d91318784748308c73dc6a164653f04ae46cd2a
2022-07-11 15:34:05 +02:00
Laurent Fainsin 4015dad491 refactor: moved the table logging in a callback
Former-commit-id: 37fa7b0da4556417ee1665d9f0375e71ce075958 [formerly 9314a3dfee09b085bb0d125d178c9f589532c6f5]
Former-commit-id: 9c8bef3f5c6560f26512bc6586f1337b9d7985f3
2022-07-11 14:41:18 +02:00
Laurent Fainsin edcc3c7bb2 feat: prerendre synthetic dataset script
Former-commit-id: f83dd74d82e9ad246a9c1ef01076f40dc1b9b897 [formerly ba0182a3220d1b2b7122227b8ef711c0a2243fc2]
Former-commit-id: 59c17c59f77419b7e420bf95617f54969eabbabc
2022-07-11 11:58:17 +02:00
Laurent Fainsin 8630f1c583 feat: some minor optimizations
Former-commit-id: 11b12e40034014a428ed542c9c1b9a89e7989b9f [formerly 8206accd8de41ce6261d30f139823804f908f35b]
Former-commit-id: 2cf3e549e6829802a8663ceab1e24ad0e7495018
2022-07-10 17:29:50 +02:00
Laurent Fainsin 8d903e7ad1 refactor: making the code actually work
Former-commit-id: 302f77ef109ca44eef8c4ce6c5b3f59ba2891884 [formerly 5ae3d84ea691440283f73b5f12e2c85b7d95b191]
Former-commit-id: 33cbfef93d9be0cefd526e1cdbd7c0435db4c613
2022-07-10 17:12:00 +02:00
Laurent Fainsin 81cbfd6212 feat: split two types of datasets
Former-commit-id: 2609316692d315f4b0df614c533bf28d20ffaf21 [formerly c1a425cb33fefa2809e591f0fe527236f6386863]
Former-commit-id: c3f96d3f272652a6162b17112be0e722c99eef57
2022-07-08 16:23:22 +02:00
Laurent Fainsin 5f46efa5a1 refactor: code splitting
Former-commit-id: 7b293e392cc7d4135ef8562faece6f491c623718 [formerly 381a418ceab2cb7f367f07b9f3ea4f3c6a41ecac]
Former-commit-id: c2ddd57c4a3592c93640170a737506bb64b60864
2022-07-08 16:06:58 +02:00
Laurent Fainsin ebb213c565 chore: renamed a notebook
Former-commit-id: 149fec5530bfa67ba2da3780e256d15a7fc1a8f1 [formerly 49e43d72aab9cdf237ac812fe51d6763465db4dd]
Former-commit-id: 04330274d840746abe66b4eb4d09c9257de3ceed
2022-07-08 13:35:24 +02:00
Laurent Fainsin 1a94fa6c09 fix: inverted logic
Former-commit-id: b3a388a527875cdc65ee3ee5c1383181bc75b6c6 [formerly 8b53736ce5adf77812831c7a3fdbca8b3b376edb]
Former-commit-id: 35b5ad1b73d5f3682fdd4251e280706373d17d36
2022-07-08 13:35:08 +02:00
Laurent Fainsin 6acac70972 feat: splitting true images in two
Former-commit-id: ad2d1d854f331672afa8f5443952151d8978e684 [formerly 79566cff12dad5a2ca4ee9b758fb7da98f652a59]
Former-commit-id: 5406367c9bdc195310b505cbcb90cbec1838f02c
2022-07-08 11:34:50 +02:00
Laurent Fainsin 164417b13f Merge branch 'pl'
Former-commit-id: 2cff12df68a22afe6b9e1133702203c4c6d86d24 [formerly 4f606e6e20e260cf94c1384ceb65c29c9ac12164]
Former-commit-id: ee4167081531b498b295c0402260da58cc4f7374
2022-07-08 11:23:02 +02:00
Laurent Fainsin a53fa4ac92 chore: testing with only real pasted sphers
Former-commit-id: 6805c289a6bb55d782fedef5233d90f985f19da3 [formerly 2c25bcbd9238002466a341bc131ff636d8d689aa]
Former-commit-id: e9de7f78afeb6c3e38b6e7cb392ac3beb25ffb0b
2022-07-08 11:21:39 +02:00
Laurent Fainsin 70b19b3b94 Merge branch 'onnx' into pl
Former-commit-id: 7f28adc38d86319181dea3e311976162ae20b6d2 [formerly 2b8801777106b1a17a9cef2c3a7d65e6ad780f3f]
Former-commit-id: 6d658bfd379005fcfdb75f97509be8dc920a8ef5
2022-07-08 11:07:26 +02:00
Laurent Fainsin 7ca448803b feat: dynamic onnx ?
Former-commit-id: 619c74a13d0674fc77bd5c1bf711013c1b3d4626 [formerly 762126125c2f108855a0837f3688f28e1002dcf7]
Former-commit-id: 7aa443fd8b68603171d2bbfa87bb9eddbe6dc066
2022-07-08 11:05:44 +02:00
Laurent Fainsin 90978bfdc3 feat: ugly training image logging
Former-commit-id: 16a25008320f436069cff9f44bf013c1c2d0f890 [formerly 683afc2cb6322ce3f1d98797b947cca8c6af09a4]
Former-commit-id: a5dae735e10107b514f028e84084ce7a303216ef
2022-07-08 09:54:45 +02:00
Laurent Fainsin 8611d8cd7a feat: better paste augmentation
Former-commit-id: 2adef7920e5f317ac3fbe0205862e29d49c2af8f [formerly 41cb0c231b00a1e992847723eb754af1a9e28eee]
Former-commit-id: f826c62f4aa3b0c9d2ea7b49f49b5839072ff259
2022-07-07 16:31:53 +02:00
Laurent Fainsin 92058da1d5 feat: fuck yeah
Former-commit-id: d472cf758e1761df9f15d1f5c7448cc4274d089f [formerly 4cd21f3ca4f0e8e22f22f61ac76e5ed4478e6937]
Former-commit-id: 9b2b0d5dd11d7b7804b1e64d52fcff2d2ea43a0b
2022-07-07 13:40:00 +02:00
Laurent Fainsin 0dd606144f feat: new paste dataset
Former-commit-id: 039874208d5a27bf01beb2746a77502fd836ae5c [formerly 66638fcabaea1044d9a2fd48e6ffb20f149ebf47]
Former-commit-id: 6bdf8bba0b3cbd8706337aa3167c36fba8855a4c
2022-07-07 12:06:41 +02:00
Laurent Fainsin b71b57285f feat: using dice_loss
feat: paste aug contrast/sharpness

Former-commit-id: 93f19e9643858a81ace14e9a697dfb6b3cca4d47 [formerly f6ef5f65e84f37b4b55a99a49442b7d30d6d3911]
Former-commit-id: 2f49a81340a91ab7456d093a849ed294457f8a83
2022-07-06 14:27:26 +02:00
Laurent Fainsin 5a74af6cdb feat: automatic batc/lr guessing sorta works
Former-commit-id: 346f1f55bab70df44bf15ab04c9a97f256e3d19c [formerly e027de4b57339dccc540ec11cfe81d5278c20d57]
Former-commit-id: 9f3537abccca7ab3d433df318cc7acf6bfc610c4
2022-07-06 11:57:21 +02:00
Laurent Fainsin f9ca8532a0 feat: multiple validation image logging
Former-commit-id: 84f7cfedb2688b15ab4401bcbfabcf6bacfa912b [formerly 51c2df31e934f80b1fd387fa123ccc8f9afac365]
Former-commit-id: 4a06d7d2891628fe798cca24f6bb10081d40e251
2022-07-06 11:01:07 +02:00
Laurent Fainsin 1306452152 fix: bad logging
Former-commit-id: 221ec6b6bfcf4e2e616a1db688ef8e93a2bb5bfc [formerly 0e7792974319cbc693fdae3597d44f9c4c196b4d]
Former-commit-id: 70f179df977715a747b981f7e2784e3c7ed88028
2022-07-05 22:52:34 +02:00
Laurent Fainsin 40ea1c3191 feat: working logging, auto_batch/lr still not working
Former-commit-id: 29d4536eb182f84eb2cc9a4e31f31bf19a4ca272 [formerly f5fd5eec9394b81f15986fb6cbabf675b2f05c04]
Former-commit-id: 3de00ee718a761221c1934b7cbaaa0ad5487856d
2022-07-05 22:31:38 +02:00
Laurent Fainsin c0772a390e feat: almost there, remove scheduler next time
Former-commit-id: 0a08b5a9559e46ca72f7d07ae84202c1412a63e9 [formerly 522877adbc8f7d132875405a86e594b4fb753850]
Former-commit-id: 2ea20809c265b8366ec2e0aa3867b13886cbd500
2022-07-05 20:50:08 +02:00
Laurent Fainsin e4562e2481 feat: kinda broken
Former-commit-id: 4cf02610721ba30c3dd1be6377daeeed907bc651 [formerly 52ef07ec8a123ddd362ac7c930eb6c915848e8b4]
Former-commit-id: 29fc18cae50625fd1f2868fc9696ca505f5648e2
2022-07-05 15:18:31 +02:00
Laurent Fainsin 36b044c719 feat: working prediction but only for 512x512
Former-commit-id: c9d88ad18de91409fc1be1f1abe59d6e75ff2235 [formerly 8bf12bad1c3e8424aa26c7bd9a441facc670b059]
Former-commit-id: 820e327f8a79bad36d1f15944012e77ba1ecd560
2022-07-05 12:17:32 +02:00
Laurent Fainsin 0fb1d4fb7a feat(WIP): broken onnx prediction
Former-commit-id: cde7623ec486cf79a710949085aadd92d8a33a3e [formerly db0f1d0b9ea536c741f23a3b683e19a9335bcd35]
Former-commit-id: 7332ccb0f74c58c3a284a4568fb8f80a6d416cf4
2022-07-05 12:06:12 +02:00
Laurent Fainsin 982dfe99d7 feat(WIP): switching to pytorch lightning
Former-commit-id: 0038dbca182717af8fc4bd846fd5be0e9fa70a9a [formerly eb5eb0717f8511bf49de8393bbdc66e727b930ff]
Former-commit-id: 540304228b146fe8e086bc4ccb770a13f84cbbcb
2022-07-04 21:40:38 +02:00
Laurent Fainsin d785a5c6be deps: pytorch-lightning
Former-commit-id: e438a57df236b38e5755777bc9b6f0b589569035 [formerly b79bdb73599a41e4c1bb20ad444538887809d0e5]
Former-commit-id: e98fc3631400ef63e1a6f658393d0dec688adb5e
2022-07-04 15:43:54 +02:00