Commit graph

30 commits

Author SHA1 Message Date
cneud
91d2a74ac9 remove redundant parentheses 2025-10-01 00:38:01 +02:00
cneud
f2f93e0251 list literal is faster than using list constructor to create a new list 2025-10-01 00:26:27 +02:00
vahidrezanezhad
f5a1d1a255 docker file to train model with desired cuda and cudnn 2025-06-25 18:24:16 +02:00
vahidrezanezhad
92954b1b7b resolving issued with saving model by steps 2025-05-05 16:13:38 +02:00
vahidrezanezhad
5694d971c5 saving model by steps is added to reading order and pixel wise segmentation use cases training 2025-05-05 15:39:05 +02:00
Clemens Neudecker
44d02687c6 Merge pull request #18 from johnlockejrr/unifying-training-models
Deprecations in train.py and check an argument in inference.py
2025-04-17 15:43:28 +02:00
vahidrezanezhad
825b2634f9 rotation augmentation is provided for machine based reading order 2025-04-16 23:36:41 +02:00
vahidrezanezhad
7b4d14b19f addinh shifting augmentation 2024-10-29 17:06:22 +01:00
johnlockejrr
f09eed1197 Changed deprecated lr to learning_rate and model.fit_generator to model.fit 2024-10-19 13:25:50 -07:00
vahidrezanezhad
a57a31673d adding foreground rgb to augmentation 2024-08-28 02:09:27 +02:00
vahidrezanezhad
aeb2ee4e3e scaling, channels shuffling, rgb background and red content added to no patch augmentation 2024-08-21 19:33:23 +02:00
vahidrezanezhad
445c45cb87 updating augmentations 2024-08-21 16:17:59 +02:00
vahidrezanezhad
22d7359db2 Transformer+CNN structure is added to vision transformer type 2024-06-12 17:39:57 +02:00
vahidrezanezhad
95faf1a4c8 transformer patch size is dynamic now. 2024-06-12 13:26:27 +02:00
vahidrezanezhad
29da23da76 binarization as a separate task of segmentation 2024-06-11 17:48:30 +02:00
vahidrezanezhad
1921e6754f updating train.py nontransformer backend 2024-06-10 22:15:30 +02:00
vahidrezanezhad
cc91e4b12c updating train.py 2024-06-07 16:24:31 +02:00
vahidrezanezhad
9638098ae7 machine based reading order training is integrated 2024-05-24 16:39:48 +02:00
vahidrezanezhad
241cb907cb Update train.py
avoid ensembling if no model weights met the threshold f1 score in the case of classification
2024-05-08 14:47:16 +02:00
vahidrezanezhad
e1f62c2e98 inference script is added 2024-05-07 13:34:03 +02:00
vahidrezanezhad
c989f7ac61 adding enhancement training 2024-05-06 18:31:48 +02:00
vahidrezanezhad
ca63c097c3 integrating first working classification training model 2024-04-29 20:59:36 +02:00
vahidrezanezhad
6e06742e66 first working update of branch 2024-04-16 01:00:48 +02:00
cneud
666a62622e code formatting with black; typos 2024-04-10 22:20:23 +02:00
cneud
39aa88669b update parameter config docs (fix #11) 2024-04-10 21:40:23 +02:00
vahidrezanezhad
57dae564b3 adjusting to tf2 2024-04-04 11:26:28 +02:00
vahid
8884b90f05 continue training, losses and etc 2021-06-22 18:47:59 -04:00
vahid
070c2e0462 first updates, padding, rotations 2021-06-22 14:20:51 -04:00
vahidrezanezhad
b5f9b9c54a Update main.py 2019-12-10 14:01:55 +01:00
979b824aa8 📝 howto: Be more verbose with the subtree pull 2019-12-09 15:33:53 +01:00