Commit Graph

17 Commits

Author SHA1 Message Date
324c690841 model and datasets adapted to new yolo format. reinforced learning is now implemented. correct_bboxes.py now capable of exporting data to train file 2023-11-16 12:19:57 +01:00
9dcf54c139 adapted data structure 2023-11-07 15:58:10 +01:00
61d5a7246b fixed the validation output. generation of folder was missing 2023-11-06 10:57:04 +01:00
a3382182e5 validation output pics now in validationaoutput/<dataset_name> 2023-10-27 11:59:01 +02:00
a6641accaf infered img not with suffix: _predicted.png 2023-10-27 09:35:29 +02:00
cc6e97c2c8 inference code copied here ... this accually was the validation step with plot output. 2023-10-27 09:33:07 +02:00
030dcfab43 changed TRAIN_DIR to DATA_DIR in all files since it is more representative in this code. test and train datasets are defined by csv files 2023-10-26 12:47:29 +02:00
4bbf3607e9 small fix 2023-10-26 11:56:14 +02:00
c2256c7001 README initialized 2023-10-26 11:54:44 +02:00
7f2d1cfb33 fixed image name problematic 2023-10-26 09:35:15 +02:00
e1a97ac493 train test split implemented using different csv files 2023-10-25 15:31:00 +02:00
ad74322f94 the model is working. however, in need to generate it and split data in test and train beforehand ... 2023-10-25 09:09:20 +02:00
8a6e7df57f generated huge dataset 2023-10-24 15:15:19 +02:00
48d84d3793 looks better ... write inference code and feed the model 2023-10-24 09:17:31 +02:00
85e675fb48 something works !!! 2023-10-24 08:56:35 +02:00
30a9e71e76 training loop runs ... no feed it 2023-10-23 15:26:39 +02:00
3f66b4a39a alot of work. dataset and loader not created 2023-10-23 14:43:01 +02:00