Commit Graph

40 Commits

Author SHA1 Message Date
Emilio Martinez a4d90691bd Minor changes on simulation 2023-07-17 18:32:32 -03:00
Emilio Martinez 1f7c602260 Add SSIM metric to logs 2023-07-17 18:31:05 -03:00
Emilio Martinez 6eeb3cca53 Fix: dataset names error handling 2023-06-26 23:54:22 -03:00
Emilio Martinez 1e7d5de144 Added bmp images for simulation 2023-06-26 23:51:37 -03:00
Emilio Martinez ed38773569 JSON based options for simulations 2023-06-26 23:29:28 -03:00
Emilio Martinez 53f8720c5b Fixed blanking issues when no VESA resolution image is given 2023-06-26 23:26:48 -03:00
Emilio Martinez 25921298c3 Updated DTutils and folder simultarion (logs, reception errors, etc) 2023-06-26 20:05:50 -03:00
Emilio Martinez acd5afa624 Added script for drunet inference on command 2023-06-19 23:33:30 -03:00
Emilio Martinez 7a5f6fc86f Changes for better training logs an visuals 2023-06-19 23:32:14 -03:00
Emilio Martinez 24a673d849 Fix (from KAIR): solved inference upsampling issue 2023-06-19 18:15:17 -03:00
Emilio Martinez 1ee3d61a2c Add skip natural patches functionalities 2023-05-13 15:20:54 -03:00
Emilio Martinez 9e9a93e834 Added epoch functionalities for training 2023-05-01 22:37:50 -03:00
Emilio Martinez d2df31cebf FIX: patches now are properly selected 2023-04-19 20:45:11 -03:00
Emilio Martinez eea43851d9 Train and logs based on epochs, not batch iteration 2023-04-19 01:47:58 -03:00
Emilio Martinez c3a0ce69fa Added support for all patches in image training 2023-04-17 22:06:48 -03:00
Emilio Martinez e07036ff4f WIP: Notebook for show training log info 2023-04-16 19:53:35 -03:00
Emilio Martinez 0b4f8234d4 Created notebook for results visualization 2023-04-16 16:25:01 -03:00
Emilio Martinez b7e1300387 Revert "Notebook with images results visualization"
This reverts commit 8d06c1b407.
2023-04-16 16:22:53 -03:00
Emilio Martinez 8d06c1b407 Notebook with images results visualization 2023-04-16 16:21:18 -03:00
Emilio Martinez 8652b0017d Added loss logs on validation set and deleted comments on drunet dataset 2023-04-16 16:18:16 -03:00
Emilio Martinez 09c1bc285f Fix: resolved validation images clipping 2023-04-07 19:37:47 -03:00
Emilio Martinez 3116e6392c Fix: changed dncnn opt to drunet opt 2023-04-07 18:21:54 -03:00
Emilio Martinez f63f68c816 Test scripts for both DnCNN and DRUNet for tempest 2023-04-06 19:31:46 -03:00
Emilio Martinez df5f0b5d32 Fix: simulated image stretch now before cropping 2023-04-02 20:02:12 -03:00
Emilio Martinez e6c0d5a79b Updated drunet-training log 2023-03-29 19:34:17 -03:00
Emilio Martinez 69aff57894 Training settings for abs value tempest 2023-03-29 19:33:45 -03:00
Emilio Martinez c00fd3efd2 Modified total variation from KAIR repo (using it as regularizer for MSE) 2023-03-29 17:09:34 -03:00
Emilio Martinez 0250ee7553 Added multiple patches per image for training (specifying num patches at train.json) 2023-03-29 17:05:44 -03:00
Emilio Martinez b60c49f319 Modified dataset for drunet (loading both GT and simulations) 2023-03-27 13:07:44 -03:00
Emilio Martinez 6972e9700f Changed noisy data folderpath 2023-03-26 23:30:43 -03:00
Emilio Martinez 92c30cdf65 Upgraded simulations using numba 2023-03-26 17:03:54 -03:00
Emilio Martinez aba3257de6 Discard .pyc and .bmp files 2023-03-26 12:24:33 -03:00
Emilio Martinez 35635b91a5 Training log (just in case) 2023-03-26 12:21:52 -03:00
Emilio Martinez a356f038a8 Added epoch parameter for training and checkpoints 2023-03-26 12:21:06 -03:00
Emilio Martinez 27b366e8e9 Fix: simulation for training dataset 2023-03-26 12:18:46 -03:00
Emilio Martínez 49b8e23f72 Added options access to total variation loss 2023-03-23 23:29:14 -03:00
Emilio Martínez e02fcbcb4a WIP: adding support to train tempest attack data 2023-03-23 23:27:58 -03:00
Emilio Martínez 1629330448 WIP: added simulation scripts (GNURadio3.8) and modifying drunet training pre-processing 2023-03-20 23:00:33 -03:00
Emilio Martínez b81bfba5bb Added KAIR and gitignore 2023-03-18 20:11:22 -03:00
Jorge Gabriel Varela Kuolas 4aaff1cfea Initial commit 2022-09-02 19:33:51 -03:00