update how it was trained
Browse files
README.md
CHANGED
@@ -43,8 +43,21 @@ dataset_val = publaynet
|
|
43 |
build_val_config = ["max_datapoints=2000"]
|
44 |
|
45 |
coco_metric = MetricRegistry.get_metric("coco")
|
46 |
-
```
|
47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
## How to fine-tune this model
|
49 |
|
50 |
To fine tune this model, please check this [Fine-tune](https://github.com/deepdoctection/deepdoctection/blob/master/notebooks/Fine_Tune.ipynb) tutorial.
|
|
|
43 |
build_val_config = ["max_datapoints=2000"]
|
44 |
|
45 |
coco_metric = MetricRegistry.get_metric("coco")
|
|
|
46 |
|
47 |
+
|
48 |
+
train_faster_rcnn(path_config_yaml=path_config_yaml,
|
49 |
+
dataset_train=dataset_train,
|
50 |
+
path_weights=path_weights,
|
51 |
+
config_overwrite=config_overwrite,
|
52 |
+
log_dir="/path/to/dir",
|
53 |
+
build_train_config=build_train_config,
|
54 |
+
dataset_val=dataset_val,
|
55 |
+
build_val_config=build_val_config,
|
56 |
+
metric=coco_metric,
|
57 |
+
pipeline_component_name="ImageLayoutService"
|
58 |
+
)
|
59 |
+
```
|
60 |
+
|
61 |
## How to fine-tune this model
|
62 |
|
63 |
To fine tune this model, please check this [Fine-tune](https://github.com/deepdoctection/deepdoctection/blob/master/notebooks/Fine_Tune.ipynb) tutorial.
|