Creates a template for config.yaml file. This specific order is preserved while saving as yaml file.
(multianimal=False)
| 39 | |
| 40 | |
| 41 | def create_config_template(multianimal=False): |
| 42 | """Creates a template for config.yaml file. |
| 43 | |
| 44 | This specific order is preserved while saving as yaml file. |
| 45 | """ |
| 46 | if multianimal: |
| 47 | yaml_str = """\ |
| 48 | # Project definitions (do not edit) |
| 49 | Task: |
| 50 | scorer: |
| 51 | date: |
| 52 | multianimalproject: |
| 53 | identity: |
| 54 | \n |
| 55 | # Project path (change when moving around) |
| 56 | project_path: |
| 57 | \n |
| 58 | # Default DeepLabCut engine to use for shuffle creation (either pytorch or tensorflow) |
| 59 | engine: pytorch |
| 60 | \n |
| 61 | # Annotation data set configuration (and individual video cropping parameters) |
| 62 | video_sets: |
| 63 | individuals: |
| 64 | uniquebodyparts: |
| 65 | multianimalbodyparts: |
| 66 | bodyparts: |
| 67 | \n |
| 68 | # Fraction of video to start/stop when extracting frames for labeling/refinement |
| 69 | start: |
| 70 | stop: |
| 71 | numframes2pick: |
| 72 | \n |
| 73 | # Plotting configuration |
| 74 | skeleton: |
| 75 | skeleton_color: |
| 76 | pcutoff: |
| 77 | dotsize: |
| 78 | alphavalue: |
| 79 | colormap: |
| 80 | \n |
| 81 | # Training,Evaluation and Analysis configuration |
| 82 | TrainingFraction: |
| 83 | iteration: |
| 84 | default_net_type: |
| 85 | default_augmenter: |
| 86 | default_track_method: |
| 87 | snapshotindex: |
| 88 | detector_snapshotindex: |
| 89 | batch_size: |
| 90 | \n |
| 91 | # Cropping Parameters (for analysis and outlier frame detection) |
| 92 | cropping: |
| 93 | #if cropping is true for analysis, then set the values here: |
| 94 | x1: |
| 95 | x2: |
| 96 | y1: |
| 97 | y2: |
| 98 | \n |