Commit graph

115 commits

Author SHA1 Message Date
Laurent Fainsin ac26b2a62a feat: switch from poetry to micromamba 2023-04-01 19:10:27 +02:00
Laurent Fainsin 2cc47bbb9e feat: cleanup dataset loaders a bit 2023-04-01 18:31:24 +02:00
Laureηt 8691735779
feat!: DETR 2023-03-27 20:51:35 +02:00
Laurent Fainsin 9d36719335 fix: dropping precision 16, unable to export in float16 2022-09-20 10:36:40 +02:00
Laurent Fainsin eb3dabe8d7 feat: ugly hack to have multiple classes 2022-09-15 10:40:04 +02:00
Laurent Fainsin 57853be03e feat: support multi image pasting 2022-09-14 16:23:19 +02:00
Laurent Fainsin c374673786 fix: dataset paths in wandb config file 2022-09-13 15:38:13 +02:00
Laurent Fainsin 41224983f7 chore: update readme
Former-commit-id: 8b5fb45bb5b3f2d77677867f78d4048475c9c60f [formerly b6a07ef688a7ae36b4326b52e0c511da1d85ff95]
Former-commit-id: add01c666a028922479909e31572097b5dbbee30
2022-09-13 11:17:42 +02:00
Laurent Fainsin 68e3b98cb2 feat: simple README.md
Former-commit-id: 59b574464d86823eab554dcd2a53b6fb63fe88a8 [formerly 27c37d0f85316e6f67eed619701707698ecca22a]
Former-commit-id: 34a00a48b3e51f88b017ae35749e6a6485050971
2022-09-12 12:19:19 +02:00
Laurent Fainsin d8f0e231e3 chore: notebook runs
Former-commit-id: f14b45aa111e91914a57511a6f93ba53be58d71f [formerly 665e94efab92ea1ab3a239a46235f912440e890b]
Former-commit-id: d4089ad9baa9d1b469b1feb0831607daced729ad
2022-09-12 11:53:28 +02:00
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