Skip to content

Commit ef16200

Browse files
Add files via upload
1 parent 74418e1 commit ef16200

1 file changed

Lines changed: 14 additions & 9 deletions

File tree

examples/image_classification_tutorial.ipynb

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
},
2424
{
2525
"cell_type": "code",
26-
"execution_count": null,
26+
"execution_count": 3,
2727
"metadata": {
2828
"colab": {
2929
"base_uri": "https://localhost:8080/",
@@ -229,12 +229,18 @@
229229
},
230230
{
231231
"cell_type": "code",
232-
"execution_count": null,
232+
"execution_count": 8,
233233
"metadata": {
234234
"scrolled": true,
235235
"tags": []
236236
},
237-
"outputs": [],
237+
"outputs": [
238+
{
239+
"output_type": "stream",
240+
"name": "stdout",
241+
"text": "(\"Dataset Statistics [{'AnnotationSet ID': 77, 'AnnotationSet name': 'Image \"\n \"classification', 'n_images': 140, 'n_classes': 3, 'n_objects': 0, \"\n \"'top_3_classes': [{'name': 'Hard-leaved pocket orchid', 'count': 60}, \"\n \"{'name': 'Canterbury bells', 'count': 40}, {'name': 'Pink primrose', \"\n \"'count': 40}], 'creation_date': None, 'last_modified_date': \"\n \"'2020-08-04T06:33:04.790286Z'}]\")\n"
242+
}
243+
],
238244
"source": [
239245
"data_stats = flowers.get_annotation_statistics()\n",
240246
"pprint(\"Dataset Statistics {}\".format(data_stats))"
@@ -290,8 +296,7 @@
290296
"A custom PyTorch ```Dataset``` object defined below is used to load data.\n",
291297
"\n",
292298
"In order to adapt this to your dataset, the following are required:\n",
293-
"\n",
294-
"- **Path to data:** Path to the Data Folder\n",
299+
"- **Path to Tags:** Path to Tags file for Train, Test, Validation split CSV generated by Remo\n",
295300
"- **Path to Annotations:** Path to Annotations CSV File (Format : file_name, class_name)\n",
296301
"- **Mapping:** Python dictionary containing mapping of class name and class index (Format : {\"class_name\" : \"class_index\"})\n",
297302
"- **transforms:** Transforms to be applied to the images before passing it to the network."
@@ -423,7 +428,7 @@
423428
"\n",
424429
"The pre-trained weights of the ```ResNet-18``` model with ImageNet are used in this tutorial.\n",
425430
"\n",
426-
"To train the model, the following details are passed to the ```train_model()``` function\n",
431+
"To train the model, the following details are to be specified.\n",
427432
"\n",
428433
"1. **Model:** The edited version of the pre-trained model.\n",
429434
"2. **Data Loaders:** The dictionary containing our training and validation dataloaders\n",
@@ -624,7 +629,7 @@
624629
"outputs": [],
625630
"source": [
626631
"classes = list(mapping.keys())\n",
627-
"flowers.create_annotation_set(\"Image Classification\", name=\"model_predictions\", path_to_annotation_file=\"./results.csv\", classes=classes)"
632+
"flowers.create_annotation_set(\"Image Classification\", name=\"model_predictions\", path_to_annotation_file=\"./results.csv\")"
628633
]
629634
},
630635
{
@@ -667,7 +672,7 @@
667672
"name": "python",
668673
"nbconvert_exporter": "python",
669674
"pygments_lexer": "ipython3",
670-
"version": "3.6.10"
675+
"version": "3.6.10-final"
671676
},
672677
"toc": {
673678
"base_numbering": 1,
@@ -934,4 +939,4 @@
934939
},
935940
"nbformat": 4,
936941
"nbformat_minor": 1
937-
}
942+
}

0 commit comments

Comments
 (0)