Skip to content

Instantly share code, notes, and snippets.

@Kartik-cl
Created May 7, 2021 08:32
Show Gist options
  • Save Kartik-cl/9faf4a9a06007bd6fb60c9d9aaa30c23 to your computer and use it in GitHub Desktop.
Save Kartik-cl/9faf4a9a06007bd6fb60c9d9aaa30c23 to your computer and use it in GitHub Desktop.
models:
- name: faster_rcnn
launchers:
- framework: dlsdk
device: CPU
adapter: ssd
inputs:
- name: im_info
type: CONST_INPUT
value: [[224, 224, 1]]
datasets:
- name: classification_dataset
data_source: <path_to_jpeg_images_for_calibration>
annotation_conversion:
converter: "voc_detection"
annotations_dir: <path_to_annotations_in_voc_format>
images_dir: <same path as data source>
imageset_file: <textfile_path with_file_names_without_extension>
has_background: true
preprocessing:
- type: resize
aspect_ratio_scale: fit_to_window
dst_height: 224
dst_width: 224
- type: padding
dst_height: 224
dst_width: 224
pad_type: right_bottom
postprocessing:
- type: faster_rcnn_postprocessing_resize
dst_height: 224
dst_width: 224
metrics:
- type: map
integral: 11point
ignore_difficult: true
presenter: print_scalar
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment