Commit graph

199 commits

Author SHA1 Message Date
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
Laurent Fainsin 55c4d05d2a fix: moved a line wrongly
Former-commit-id: 93f2c7ee9a1926248716328dc9dd5a46395373be [formerly 598da90e69422bb29bf0c2770b789773c6360981]
Former-commit-id: 354585b269180f4785f8bd05164eb8c8b0d16dba
2022-07-04 15:38:49 +02:00
Laurent Fainsin c69579f9da fix: retyped the deleted inital logging
Former-commit-id: 219b4a406826f90145d6806d1f1d438d9bd44282 [formerly abe23af73cef108a97791ee21218f7a121f60ac2]
Former-commit-id: c1f0d82783ec3ee4ee1518c10d2b9a9c91e6ec26
2022-07-04 14:50:29 +02:00
Laurent Fainsin 0d6f85518e feat: ajout des données de test
Former-commit-id: c582ae71d296afe90d25127f541c696052172a2a [formerly b5fe53254d424e3d6ea74573378a716ccd429d84]
Former-commit-id: f0ae70c1025d70af43b0f172e5abaeeba819999f
2022-07-04 14:38:48 +02:00
Laurent Fainsin 2571e5c6d3 feat: validate every 100 steps
Former-commit-id: a3367de4ed56c5a708d66e7cd6be27d52bb92ccc [formerly 4606a91526eae57d56fc93df7ed34b867495e1c5]
Former-commit-id: 6584449cd25b18ddd46f6804c8b1653e1c72dda0
2022-07-01 15:31:53 +02:00
Laurent Fainsin cf8f52735a feat: log learning_rate
Former-commit-id: aaf6be4efe43d65e70650ee8c07b81b584a8d70e [formerly c4289255d70c75c72b684886824832ab61df533b]
Former-commit-id: a163c42fa2ca66e32c093424ed8ffdc3b82b5ea5
2022-07-01 14:32:30 +02:00
Laurent Fainsin d839aec1af feat: combine multiple datasets
Former-commit-id: bec67c2948227cdbfe05bb8f69e8083f7591e965 [formerly b24d77ef7c42456893a0b797e0a6c925a2402d08]
Former-commit-id: 391e17bf37e44151215179cde2d157f8e5bb5182
2022-07-01 12:00:25 +02:00
Laurent Fainsin beafa768f7 feat: display binary mask on top of prediction
Former-commit-id: a81b30919c6ef0998617822fd57e54209dcb2cfa [formerly 9f9cf75289c5f731dc6c93f86dff237aed88722c]
Former-commit-id: d7272aefcf16e2e22081811c0452d7cb7e0980f3
2022-07-01 11:19:15 +02:00
Laurent Fainsin 31ceb97996 fix: removed now useless Sequential
Former-commit-id: 4b97d6d3bef3814f7cfc6c6ceca7833105b4211a [formerly 29f131cf14a73b22427f4ef2318def6870cbe7e0]
Former-commit-id: db11e7b2ef2ba39fe11e69251731ea2489d3be31
2022-07-01 10:33:35 +02:00
Laurent Fainsin 2ab95734e4 feat: even more wandb logging
Former-commit-id: 1a3c28040a734ca2229e33603405054abc8e3000 [formerly 907e4f7cae3c25a84baf0eaa5ec4d03ddaea0bdb]
Former-commit-id: fdfb7dcb7d0573efbff79956e7a4bebfe26e2171
2022-07-01 10:27:12 +02:00
Laurent Fainsin 7bdac6583b feat: binarized the masks + lots of new metrics to fix
Former-commit-id: c840d14f722503d241f6bb6d899630ad6345aca0 [formerly e435a21234620add4f0e4e269a4141e5c1508cd9]
Former-commit-id: 8006af185fd68cc88b2305a02513106c16758d77
2022-06-30 23:28:38 +02:00
Laurent Fainsin e20a989c41 feat: added some training metrics
Former-commit-id: 05f9c9c44914cc2ef3443791e71d2f9ca893751b [formerly 0cec2187f92c52328f07cca65ebaeb7ab3641ac3]
Former-commit-id: 4749dbea8fd69798e08e34a91156d3111c7504a8
2022-06-30 21:26:12 +02:00
Laurent Fainsin d9f2dc2bfb feat: got positive loss values !
Former-commit-id: 84e2a715b843ecee2e12e4878fcee4a52bb0a4cb [formerly 1a5fc82bc099885853b7b4deff81b779dafd0168]
Former-commit-id: c82cd66d6c432555a126e506631dfa2fd756437e
2022-06-30 16:47:28 +02:00
Laurent Fainsin 3e335fbcb5 fix: bad loop condition
Former-commit-id: 71677d5f1a37b742985f4335c6dc1c476a82ba48 [formerly df8e6e36e785f07970d47be9183bbab5ac386e01]
Former-commit-id: 1162f7d0a8eac822fbd2e3f38edd2f2c8a81124f
2022-06-30 15:54:36 +02:00
Laurent Fainsin 3056ebc498 fix: typo
Former-commit-id: 0336f7e3a3809f6501dec4ca11a9ae1f3be44f29 [formerly 1bd5e48616d48d52e13067ff505a94162acf9e6f]
Former-commit-id: 471e9d11a26b14f6b878f9735d3e9c40eb10161b
2022-06-30 14:43:41 +02:00
Laurent Fainsin 24df16a612 f5
Former-commit-id: 4207d1193b6eadcb491a72f51f2c512150f774c3 [formerly cc23543f39cfd6e0894d12d18cd33a18d2b4a20f]
Former-commit-id: 23b4c151217a9db6b9e21105b1e0954ec2f78ce1
2022-06-30 14:36:48 +02:00
Laurent Fainsin 8c9ed80c6a feat: use wandb config instead of args
Former-commit-id: ffb1cb9a6e978c41b3b62388c657ccdb13c4ad67 [formerly d557639e5a203e2ba44ebcf4466c42074f215fa0]
Former-commit-id: 0d3dd6a81a66348fd4caa840a2727680554854f3
2022-06-30 14:04:02 +02:00
Laurent Fainsin f4ed2f799e revert: want back to table display for predictions
Former-commit-id: 2af16f486ffeaca232aee35d360fe75e78adfc9e [formerly e072ee4b33587b45f6c6c34f809b9638a9e0b569]
Former-commit-id: 69c874da0fbe6e2a5cb2421fa20ffb9c34b36fc1
2022-06-30 11:44:39 +02:00
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