Commit Graph
486 Commits
Author SHA1 Message Date
brian d6036a95a3 fixed deprecated import 2024-10-19 20:09:32 +00:00
brian 14ccaace95 mypy fixed types 2024-10-19 20:03:54 +00:00
brian 6d327bee22 fixed type bug 2024-10-19 19:55:58 +00:00
brian 270dc8e4d1 fixed type 2024-10-19 19:54:40 +00:00
brian e49a0c37a6 ensure env var set 2024-10-19 19:45:42 +00:00
brian a0726857a0 ensure env var set 2024-10-19 19:45:16 +00:00
brian bf44947024 added utility script
CI Pipeline / Test (pull_request) Failing after 2m31s
CI Pipeline / Build and Publish (./Dockerfile.model, ${{ vars.docker_repo_url }}/${{ gitea.repository }}/model) (pull_request) Has been skipped
CI Pipeline / Build and Publish (./Dockerfile.web_ui, ${{ vars.docker_repo_url }}/${{ gitea.repository }}/web_ui) (pull_request) Has been skipped
2024-10-19 19:15:41 +00:00
brian 4f22b9755c updated import 2024-10-19 19:15:20 +00:00
brian ae3ac714d4 added placeholder files for training and validation files 2024-10-19 19:14:56 +00:00
brian ad069269ac added settings for resnet18-based model 2024-10-19 19:14:34 +00:00
brian a54e7bbbb4 installed packages 2024-10-19 19:13:45 +00:00
brian 1f62054f36 installed packages 2024-10-19 19:13:35 +00:00
brian 59048a6862 added model training script 2024-10-19 19:12:21 +00:00
brian a4764b8fc5 added ignore for training run folder 2024-10-19 19:11:02 +00:00
brian bf4cc9c483 ensure non-empty input 2024-10-19 19:10:33 +00:00
brian f35cdefc21 removed deprecated dependency 2024-10-19 19:08:51 +00:00
Brian Bjarke Jensen 493da70ba4 updated to do prediction and print result 2024-07-28 23:22:40 +02:00
Brian Bjarke Jensen b415137e25 added method to instantiate from prediction dict 2024-07-28 23:19:51 +02:00
Brian Bjarke Jensen cead3d24c8 added from_tensor method 2024-07-28 23:13:15 +02:00
Brian Bjarke Jensen 0bfbb1920c fixed conversion of classes 2024-07-28 23:12:06 +02:00
Brian Bjarke Jensen 164102557e removed unused import 2024-07-28 23:10:15 +02:00
Brian Bjarke Jensen cf923a228d updated model name reference 2024-07-28 21:37:28 +02:00
Brian Bjarke Jensen 9f4a09a18c fixed definition of layers 2024-07-28 21:36:21 +02:00
Brian Bjarke Jensen 849b59e575 removed double connection to minio 2024-07-28 21:34:56 +02:00
Brian Bjarke Jensen 4a7e33b527 updated referenced model name 2024-07-28 21:32:16 +02:00
Brian Bjarke Jensen e5fad6d765 added function to generate and save new model 2024-07-28 21:31:48 +02:00
Brian Bjarke Jensen be6833ac6a added option to download weights 2024-07-28 21:25:24 +02:00
Brian Bjarke Jensen 6541a47668 added option to download resnet18 weights 2024-07-28 21:19:47 +02:00
Brian Bjarke Jensen 96ccdb5c90 moved file for clarity 2024-07-28 21:16:32 +02:00
Brian Bjarke Jensen 6e9c0a2626 fixed number of inputs 2024-07-28 21:13:51 +02:00
Brian Bjarke Jensen 396bf29f32 updated function name 2024-07-28 21:08:51 +02:00
Brian Bjarke Jensen 05774364fc renamed function for clarity 2024-07-28 21:08:27 +02:00
Brian Bjarke Jensen 74a9538feb renamed function for clarity 2024-07-28 21:05:15 +02:00
Brian Bjarke Jensen 9423c3378a fixed dockerfile syntax 2024-07-28 21:02:59 +02:00
Brian Bjarke Jensen e6fdbfe0b0 changed to not download pretrained weights for ResNet18 2024-07-28 21:01:19 +02:00
Brian Bjarke Jensen 6c354b23bd test run prediction on image 2024-07-27 21:41:29 +02:00
Brian Bjarke Jensen 3ad6c161b1 added new modules 2024-07-27 21:40:38 +02:00
Brian Bjarke Jensen a1ff6a6f33 moved for clarity 2024-07-27 21:40:17 +02:00
Brian Bjarke Jensen cf4f67d7a5 installed tqdm 2024-07-27 21:38:18 +02:00
Brian Bjarke Jensen 7c968db8ac renamed file for clarity 2024-07-27 21:18:39 +02:00
Brian Bjarke Jensen e656cd4e12 remove unneeded code for running in docker container 2024-07-27 21:08:47 +02:00
Brian Bjarke Jensen 8b1893b30a implemented load_model function 2024-07-24 09:13:28 +02:00
Brian Bjarke Jensen 96d868a40c added function to load model 2024-07-24 09:12:07 +02:00
Brian Bjarke Jensen e698d058ff removed debug log 2024-07-24 09:11:01 +02:00
Brian Bjarke Jensen 8aafbdae6c fixed hardcoded path 2024-07-24 09:10:37 +02:00
Brian Bjarke Jensen ec4162e34d removed newline char 2024-07-22 23:50:55 +02:00
Brian Bjarke Jensen 99b7067cfe added container resource limits 2024-07-22 23:48:01 +02:00
Brian Bjarke Jensen e62435cf22 added full model load script 2024-07-22 23:47:39 +02:00
Brian Bjarke Jensen 3560a61277 added model name loader 2024-07-22 23:47:11 +02:00
Brian Bjarke Jensen 7f0cbea937 added model name link file 2024-07-22 23:46:15 +02:00