|
66 | 66 | "- RetinaNet\n",
|
67 | 67 | "- FasterRCNN\n",
|
68 | 68 | "- YOLOv3\n",
|
69 |
| - "- MMDetection:\n", |
70 |
| - " - atss\n", |
71 |
| - " - carafe\n", |
72 |
| - " - cascade_rcnn\n", |
73 |
| - " - cascade_rpn\n", |
74 |
| - " - dcn\n", |
| 69 | + "- ATSS\n", |
| 70 | + "- CARAFE\n", |
| 71 | + "- CascadeRCNN\n", |
| 72 | + "- CascadeRPN\n", |
| 73 | + "- DCN\n", |
75 | 74 | "\n",
|
76 | 75 | "**Pixel Classification**\n",
|
77 | 76 | "\n",
|
78 | 77 | "- DeepLab\n",
|
79 | 78 | "- UnetClassifier\n",
|
80 | 79 | "- PSPNetClassifier\n",
|
81 |
| - " - ann\n", |
82 |
| - " - apcnet\n", |
83 |
| - " - ccnet\n", |
84 |
| - " - cgnet\n", |
85 |
| - " - deeplabv3\n" |
| 80 | + "- ANN\n", |
| 81 | + "- APCNet\n", |
| 82 | + "- CCNet\n", |
| 83 | + "- CGNet\n", |
| 84 | + "- HRNet\n" |
86 | 85 | ]
|
87 | 86 | },
|
88 | 87 | {
|
|
166 | 165 | " - _advanced_ : To be used when the user also wants to tune hyper parameters of the two best performing models from the basic mode.\n",
|
167 | 166 | " \n",
|
168 | 167 | "\n",
|
169 |
| - "- `networks` (Optional Parameter):\n", |
170 |
| - " The list of models that will be used in the training process. If the user does not provide the parameter value, the `AutoDL` class selects all of the supported networks, however the user can select one or more networks by passing the network names in a list.\n", |
| 168 | + "- `network` (Optional Parameter):\n", |
| 169 | + " The list of models that will be used in the training process. If the user does not provide the parameter value, the `AutoDL` class selects all of the supported networks, however the user can select one or more networks by passing the network names as string in a list.\n", |
171 | 170 | "\n",
|
172 | 171 | " - _Supported Object Detection models_\n",
|
173 |
| - " - SingleShotDetector, RetinaNet, FasterRCNN, YOLOv3, MMDetection\n", |
| 172 | + " - SingleShotDetector, RetinaNet, FasterRCNN, YOLOv3, ATSS, CARAFE, CascadeRCNN, CascadeRPN, DCN\n", |
174 | 173 | " - _Supported Object Detection models_\n",
|
175 |
| - " - DeepLab, UnetClassifier, PSPNetClassifier, MMSegmentation\n", |
| 174 | + " - DeepLab, UnetClassifier, PSPNetClassifier, ANN, APCNet, CCNet, CGNet, HRNet\n", |
176 | 175 | " \n",
|
177 | 176 | "\n",
|
178 | 177 | "- `verbose` (Optional Parameter): Optional Boolean.\n",
|
|
539 | 538 | ],
|
540 | 539 | "metadata": {
|
541 | 540 | "kernelspec": {
|
542 |
| - "display_name": "Python 3", |
| 541 | + "display_name": "Python 3 (ipykernel)", |
543 | 542 | "language": "python",
|
544 | 543 | "name": "python3"
|
545 | 544 | },
|
|
553 | 552 | "name": "python",
|
554 | 553 | "nbconvert_exporter": "python",
|
555 | 554 | "pygments_lexer": "ipython3",
|
556 |
| - "version": "3.9.11" |
| 555 | + "version": "3.9.15" |
557 | 556 | }
|
558 | 557 | },
|
559 | 558 | "nbformat": 4,
|
|
0 commit comments