Loading README.md +5 −0 Original line number Diff line number Diff line Loading @@ -44,6 +44,7 @@ python3 retrain.py \ --image_dir /softpro/ss18/caption/cocoapi/train_all_objects/ ``` The directory has to contain subdirectories which have the name of the label and contain all image which are tagged with the label. You can also pass different hyperparamters as arguments: ``` python3 retrain.py \ Loading Loading @@ -109,6 +110,10 @@ python retrain.py \ --image_dir=/softpro/ss18/caption/cocoapi/multilabel_outer/ ``` The image directory this time contains all the training images without subdirectories. Additionally you have to open the the retrain.py and change the path of IMAGE_LABELS_DIR and ALL_LABELS_FILE. IMAGE_LABELS_DIR is a directory which contains a .txt file for every image describing all labels on that image. ALL_LABELS_FILE is a .txt file that contains all labels. * **Results:** [Trained model (default)](/models/multilabel_default/) Loading Loading
README.md +5 −0 Original line number Diff line number Diff line Loading @@ -44,6 +44,7 @@ python3 retrain.py \ --image_dir /softpro/ss18/caption/cocoapi/train_all_objects/ ``` The directory has to contain subdirectories which have the name of the label and contain all image which are tagged with the label. You can also pass different hyperparamters as arguments: ``` python3 retrain.py \ Loading Loading @@ -109,6 +110,10 @@ python retrain.py \ --image_dir=/softpro/ss18/caption/cocoapi/multilabel_outer/ ``` The image directory this time contains all the training images without subdirectories. Additionally you have to open the the retrain.py and change the path of IMAGE_LABELS_DIR and ALL_LABELS_FILE. IMAGE_LABELS_DIR is a directory which contains a .txt file for every image describing all labels on that image. ALL_LABELS_FILE is a .txt file that contains all labels. * **Results:** [Trained model (default)](/models/multilabel_default/) Loading
