Commit graph

105 commits

Author SHA1 Message Date
Laurent Fainsin 1ed1bb185c feat: more docstrings and typing
Former-commit-id: 9082187ec9d66e93c0195374022290cb9231be00 [formerly c55aabb212241973372630e9a078da5fc0342abf]
Former-commit-id: ee4dbe5392b99e03db27da2873c180c84a75737f
2022-09-12 11:45:19 +02:00
Laurent Fainsin 185060d469 feat: docstrings + typing
Former-commit-id: 38dfddce97808be4077aa7d943f34096429bc49c [formerly e5d46bbdd39dbd70b65f8deb5fe9ad66d0ecd9b0]
Former-commit-id: 35f5b2cd55cc38756e03c013ee705b137b425239
2022-09-12 10:59:37 +02:00
Laurent Fainsin 85c2febcac chore: bunch of small stuff
Former-commit-id: e6eea69309c723face4c1f09ad935451ce715eee [formerly f59c07f943f0cddc6db2a0512923f2960a2400bd]
Former-commit-id: 0e2f44ec2b051ca2b31fe16c59e6702a6890701c
2022-09-12 09:28:29 +02:00
Laurent Fainsin b701afe363 feat: better image logging
Former-commit-id: 5ce04a5534cef72a3815be13cd79731800b7419f [formerly 47ab70b94c3f4a696b4e5a131e087660dba0b8ba]
Former-commit-id: 81858cf5970cdecdbd0a151e3547a25857e3e958
2022-09-12 09:28:11 +02:00
Laurent Fainsin 0693f02d83 feat: create new datasets for generation and loading of pre rendered images
Former-commit-id: 4ccd97c3583c7def1e5e988a5beb8fcda7545fc9 [formerly 6a7e1eb28ade30f3e4b76337c209ac7e5b3b1cbf]
Former-commit-id: 8a552d696c6e47378a36a8680e3a09d151de25f1
2022-09-12 09:27:49 +02:00
Laurent Fainsin c6c08ac98a fix: limit box detections per images to 10
Former-commit-id: 3b49b7b2c9d5093ab989e8757c6b0429eb9746bd [formerly abf1897a28a7591c6fdf9cf51b9c832ca48cc10b]
Former-commit-id: c1db1fe1e9475adbdf10f067db9206d69fe5ed2f
2022-09-12 09:26:42 +02:00
Laurent Fainsin f50b758102 feat: compute mAP for both "bbox" and "segm"
Former-commit-id: 85379c46352b725ceaa7955d7d44b350ef02708a [formerly 2a503dbc4852efc4d21284ded746451ba9aaa495]
Former-commit-id: b8a9c3e26b4085b0724b17e79bfa30b7727fb310
2022-09-12 09:25:40 +02:00
Laurent Fainsin 5fe7ceb306 chore: move notebooks in parant folder
Former-commit-id: 56075c7a86305a7a9644dfaacd5bdcb62a863baf [formerly 35967dee0a3c159669b50e2823f36ff25e036f1e]
Former-commit-id: b98a086c0ea8970562055cc3ad1a025a77b94b35
2022-09-07 14:34:32 +02:00
Laurent Fainsin e94ea4147c feat: use LinearWarmupCosineAnnealingLR
Former-commit-id: a7292fe2b0898513fd0e913c2ae352a187f05b12 [formerly 432664f5bd6c3f8f54c221e4d7cc8853d08ea55b]
Former-commit-id: b777bd5053a005ec9b4da7db19a4dbe2a1ba41fd
2022-09-07 10:44:29 +02:00
Laurent Fainsin 291ea632bd feat: bunch of new callbacks
Former-commit-id: 40fa0eaabd63cc00becb79164255289b18faf26f [formerly 2e813e0fbd9b7316d500b1d3f694b680d1e4e949]
Former-commit-id: 8b0372f2a4d3657e0728f52ae12c529df6985a07
2022-09-07 10:44:10 +02:00
Laurent Fainsin db90446eb0 chore: remove unused file
Former-commit-id: bf466229627337e31bff835bd7615b764ab982ac [formerly 919ecf8585bd9910ffd9722ff390e16d265dc2fa]
Former-commit-id: 2e4531522077a9191c55b5952f05bd0c4f2346f4
2022-09-07 10:43:22 +02:00
Laurent Fainsin 3bbc132359 fix: augmentations
Former-commit-id: d06568c40226cb9b2e4f80b4fe1dea1f094a5934 [formerly ecc833dfd23fc9065637e24de9b6ce94410f5c10]
Former-commit-id: 70d75d9c26a3010618a5b6e42ca197d1917ef640
2022-09-07 10:43:12 +02:00
Laurent Fainsin 56e24615e3 fix: TableLog
Former-commit-id: 9ec175df257d4f969c3b024735a021197f070432 [formerly 96dfc4589152e497cfdc5d0f9291957307d2cda4]
Former-commit-id: 2f7899f654158e5be4b4ae59db0cc524d1a8ead4
2022-09-07 10:42:58 +02:00
Laurent Fainsin c312513eff feat: checkpointing + torchmetrics + ...
Former-commit-id: 781a11f646964567b3bc6831caa5e380748b84e1 [formerly 7ea5f37ecb7fd5c95e5d1b7b1aa899d986ecff2a]
Former-commit-id: 06974fa163a6a8eb881d5981f9c6debe63f1b4bb
2022-09-05 16:04:52 +02:00
Laurent Fainsin c0a52196fd fix: remove execution permission
Former-commit-id: 3fcf272c3fca5776faa8baeded0f20035ad1d472 [formerly fd82c97d3c003ad2010e64ba58d6badba91c1c2a]
Former-commit-id: 683cfc02f1c776b8c15e46c234984b72608b3066
2022-09-02 16:34:00 +02:00
Laurent Fainsin e1e74242de refactor: remove pycocotools
Former-commit-id: ebf5f3b85c152491ef299d4d2773c96b8ff3d394 [formerly d6923f15603da3302224199a0b8404c83b091bca]
Former-commit-id: 1d6ecdd7fffe5bd7f77e2af3d0d4d9b8df48ae53
2022-09-02 15:56:59 +02:00
Laurent Fainsin d65dc7b136 feat: dataset augmentation
Former-commit-id: 6f26e70e716f08b8ae3c92cf9693d00a5e67a9a6 [formerly 0d59c038239feb6617a0af75aa32247064dabfdc]
Former-commit-id: 3a3b32107a90986d3f2580d375ceef1e2f1c2256
2022-09-02 14:34:18 +02:00
Laurent Fainsin ca125bec0c feat: POC onnxruntime with mask R-CNN
Former-commit-id: e10c3a9f332145797979ce1ef556909a8f76c2f3 [formerly 625ae128a1fe6dbc0eec8da170fd0b5390128fe2]
Former-commit-id: 5dd429a665b96f5a833ca7734815d771600241b3
2022-08-30 13:23:17 +02:00
Laurent Fainsin 562ef110af feat: basic validation
Former-commit-id: 9d6ac1da57e613d3fa4db7ba68560509611c2b79 [formerly f588a5a0781e7375baa6ce5a4723a7b0d557316a]
Former-commit-id: 7415ee71e570da13f9de361131156ba0d2131d8a
2022-08-30 10:20:32 +02:00
Laurent Fainsin 04ddf75dd8 chore: bettered predict.ipynb
Former-commit-id: 375b2fa79b8b94f815f261598592d11d2ac8a92d [formerly 547c360afa04fc6986831422ce861a8470563c00]
Former-commit-id: 2915fc814fcf760ee4af9fb2447fe4f1cf163d2e
2022-08-30 10:18:42 +02:00
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