YOWO.yaml 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. Global:
  2. checkpoints: null
  3. pretrained_model: https://videotag.bj.bcebos.com/PaddleVideo-release2.3/YOWO_epoch_00005.pdparams
  4. output_dir: ./output/
  5. device: gpu
  6. use_visualdl: False
  7. save_inference_dir: ./inference
  8. # training model under @to_static
  9. to_static: False
  10. algorithm: YOWO
  11. MODEL: #MODEL field
  12. framework: "YOWOLocalizer" #Mandatory, indicate the type of network, associate to the 'paddlevideo/modeling/framework/' .
  13. backbone: #Mandatory, indicate the type of backbone, associate to the 'paddlevideo/modeling/backbones/' .
  14. name: "YOWO" #Mandatory, The name of backbone.
  15. num_class: 24
  16. loss:
  17. name: "RegionLoss"
  18. num_classes: 24
  19. num_anchors: 5
  20. anchors: [0.70458, 1.18803, 1.26654, 2.55121, 1.59382, 4.08321, 2.30548, 4.94180, 3.52332, 5.91979]
  21. object_scale: 5
  22. noobject_scale: 1
  23. class_scale: 1
  24. coord_scale: 1
  25. DATASET: #DATASET field
  26. batch_size: 8 #Mandatory, bacth size
  27. num_workers: 4 #Mandatory, XXX the number of subprocess on each GPU.
  28. test_batch_size: 4
  29. valid_batch_size: 4
  30. train:
  31. format: "UCF24Dataset" #Mandatory, indicate the type of dataset, associate to the 'paddlevidel/loader/dateset'
  32. image_dir: "data/ucf24" #Mandatory, raw data folder path
  33. file_path: "data/ucf24/trainlist.txt" #Mandatory, train data index file path
  34. valid:
  35. format: "UCF24Dataset" #Mandatory, indicate the type of dataset, associate to the 'paddlevidel/loader/dateset'
  36. image_dir: "data/ucf24" #Mandatory, raw data folder path
  37. file_path: "data/ucf24/testlist.txt" #Mandatory, test data index file path
  38. test:
  39. format: "UCF24Dataset" #Mandatory, indicate the type of dataset, associate to the 'paddlevidel/loader/dateset'
  40. image_dir: "data/ucf24"
  41. file_path: "data/ucf24/testlist.txt" #Mandatory, test data index file path
  42. PIPELINE: #PIPELINE field TODO.....
  43. train: #Mandotary, indicate the pipeline to deal with the training data, associate to the 'paddlevideo/loader/pipelines/'
  44. sample:
  45. name: "SamplerUCF24"
  46. num_frames: 16
  47. valid_mode: False
  48. transform: #Mandotary, image transform operator.
  49. - YowoAug:
  50. valid_mode: False
  51. valid: #Mandotary, indicate the pipeline to deal with the training data, associate to the 'paddlevideo/loader/pipelines/'
  52. sample:
  53. name: "SamplerUCF24"
  54. num_frames: 16
  55. valid_mode: True
  56. transform: #Mandotary, image transform operator.
  57. - YowoAug:
  58. valid_mode: True
  59. test:
  60. sample:
  61. name: "SamplerUCF24"
  62. num_frames: 16
  63. valid_mode: True
  64. transform: #Mandotary, image transform operator.
  65. - YowoAug:
  66. valid_mode: True
  67. OPTIMIZER: #OPTIMIZER field
  68. name: Adam
  69. learning_rate:
  70. learning_rate: 0.0001
  71. name: 'MultiStepDecay'
  72. milestones: [1, 2, 3, 4]
  73. gamma: 0.5
  74. weight_decay:
  75. name: "L2"
  76. value: 0.0005
  77. GRADIENT_ACCUMULATION:
  78. global_batch_size: 128 # Specify the sum of batches to be calculated by all GPUs
  79. METRIC:
  80. name: 'YOWOMetric'
  81. gt_folder: 'data/ucf24/groundtruths_ucf'
  82. result_path: 'output/detections_test'
  83. threshold: 0.5
  84. log_interval: 100
  85. for_paddlex: True
  86. INFERENCE:
  87. name: 'YOWO_Inference_helper'
  88. num_seg: 16
  89. target_size: 224
  90. Infer:
  91. transforms:
  92. - ReadVideo:
  93. num_seg: 16
  94. - ResizeVideo:
  95. target_size: 224
  96. - Image2Array:
  97. data_format: 'tchw'
  98. - NormalizeVideo:
  99. scale: 255.0
  100. PostProcess:
  101. - DetVideoPostProcess:
  102. nms_thresh: 0.5
  103. score_thresh: 0.4
  104. label_list:
  105. - Basketball
  106. - BasketballDunk
  107. - Biking
  108. - CliffDiving
  109. - CricketBowling
  110. - Diving
  111. - Fencing
  112. - FloorGymnastics
  113. - GolfSwing
  114. - HorseRiding
  115. - IceDancing
  116. - LongJump
  117. - PoleVault
  118. - RopeClimbing
  119. - SalsaSpin
  120. - SkateBoarding
  121. - Skiing
  122. - Skijet
  123. - SoccerJuggling
  124. - Surfing
  125. - TennisSwing
  126. - TrampolineJumping
  127. - VolleyballSpiking
  128. - WalkingWithDog
  129. model_name: "YOWO"
  130. log_interval: 20 #Optional, the interval of logger, default:10
  131. save_interval: 1
  132. epochs: 5 #Mandatory, total epoch
  133. log_level: "INFO" #Optional, the logger level. default: "INFO"
  134. val_interval: 1
  135. label_dict_path: data/ucf24/label_map.txt