Commit 0f7ee791 authored by holzinger's avatar holzinger
Browse files

Update README.md

parent 545314ae
Loading
Loading
Loading
Loading
+45 −0
Original line number Diff line number Diff line
@@ -29,6 +29,51 @@ python setup.py install
  
[Coco Preprocessing Code](/preprocessing/coco_preprocessing/)

#### Reqired Input

The following folder structure is required to run the retraining process.

#### Before Preprocessing:

Assumption: Your pictures from coco is in jpg files in the `cocoapi/images/train/` subfolder and the coco annotation files are in the `cocoapi/annotations/train/` subfolder.

```
cocoapi/                    
    images/
        train/
            *.jpg
    annotations/
        train/
            instances_train2017.json
            captions_train2017.json
```

#### After Preprocessing:

* [exract_multilabel_pics.py](/preprocessing/coco_preprocessing/extract_multilabel_pics.py) will create the following folders and files (images for multilabel training):
    
    ```
    cocoapi/
        multilabel_outer/
            multilabel_images/
                *.jpg
    ```

* [categories_of_pictures.py](/preprocessing/coco_preprocessing/categories_of_pictures.py) will create the following folders and files (labels for multilabel training):

    ```
    cocoapi/
        images_txt/
            *.txt
    ```
    
* [categories_labels.py](/preprocessing/coco_preprocessing/categories_labels.py) will create the following file (categories of coco):

    ```
    cocoapi/
        labels.txt
    ```

#### Retraining

Tensorflow pretrained image classification models available at [https://github.com/tensorflow/models/tree/master/research/slim](https://github.com/tensorflow/models/tree/master/research/slim)