|
|
@@ -134,3 +134,42 @@ def main(args):
|
|
|
|
|
|
main(parser.parse_args())
|
|
|
```
|
|
|
+This example can be run with
|
|
|
+```bash
|
|
|
+python example.py data.yml CUB200 GLOBAL -mt cvmodelz.ResNet50 -g0 --input_size infer_big --batch_size 24
|
|
|
+```
|
|
|
+using this data config (`data.yml`):
|
|
|
+
|
|
|
+```yaml
|
|
|
+BASE_DIR: /home/korsch_data/
|
|
|
+
|
|
|
+DATA_DIR: datasets
|
|
|
+MODEL_DIR: models
|
|
|
+
|
|
|
+MODELS:
|
|
|
+ cvmodelz.InceptionV3: &inception
|
|
|
+ folder: inception
|
|
|
+ weights:
|
|
|
+ inat: model.inat.ckpt.npz
|
|
|
+ imagenet: model.imagenet.ckpt.npz
|
|
|
+
|
|
|
+ cvmodelz.ResNet50: &resnet50
|
|
|
+ folder: resnet
|
|
|
+ weights:
|
|
|
+ imagenet: model.npz
|
|
|
+
|
|
|
+DATASETS:
|
|
|
+ CUB200: &cub200
|
|
|
+ folder: birds/cub200
|
|
|
+ annotations: "ORIGINAL"
|
|
|
+ annotation_type: FILE_LIST
|
|
|
+ n_classes: 200
|
|
|
+
|
|
|
+PART_TYPES:
|
|
|
+ GLOBAL: &parts_global
|
|
|
+ feature_suffix: ""
|
|
|
+
|
|
|
+PARTS: {}
|
|
|
+```
|
|
|
+
|
|
|
+
|
