Fix some config and add description

pull/52/head
huanghaian 2022-09-21 10:12:00 +08:00
parent b59aa0fb5a
commit c683d1f94d
5 changed files with 10 additions and 11 deletions

View File

@ -20,9 +20,11 @@ YOLOv5 is a family of object detection architectures and models pretrained on th
**Note**:
1. `fast` means that `YOLOv5DetDataPreprocessor` and `yolov5_collate` are used for data preprocessing, which is faster for training, but less flexible for multitasking. Recommended to use fast version config if you only care about object detection.
2. `SyncBN` means use SyncBN, `AMP` indicates training with mixed precision.
3. We use 8x A100 for training, and the single-GPU batch size is 16. This is different from the official code.
4. The performance is unstable and may fluctuate by about 0.4 mAP.
2. `detect` means that the network input is fixed to `640x640` and the post-processing thresholds is modified.
3. `SyncBN` means use SyncBN, `AMP` indicates training with mixed precision.
4. We use 8x A100 for training, and the single-GPU batch size is 16. This is different from the official code.
5. The performance is unstable and may fluctuate by about 0.4 mAP.
6. `balloon` means that this is a demo configuration.
## Citation

View File

@ -16,7 +16,8 @@ test_pipeline = [
'scale_factor', 'pad_param'))
]
val_dataloader = dict(dataset=dict(pipeline=test_pipeline))
val_dataloader = dict(
dataset=dict(pipeline=test_pipeline, batch_shapes_cfg=None))
test_dataloader = val_dataloader
model = dict(

View File

@ -188,12 +188,7 @@ val_dataloader = dict(
data_prefix=dict(img='val2017/'),
ann_file='annotations/instances_val2017.json',
pipeline=test_pipeline,
batch_shapes_cfg=dict(
type='BatchShapePolicy',
batch_size=val_batch_size_per_gpu,
img_size=img_scale[0],
size_divisor=32,
extra_pad_ratio=0.5)))
batch_shapes_cfg=batch_shapes_cfg))
test_dataloader = val_dataloader

View File

@ -3,6 +3,7 @@
## YOLO series model basic class
The structural Graph is provided by RangeKing@GitHub. Thank you RangeKing
<div align=center>
<img src="https://user-images.githubusercontent.com/27466624/190986949-01414a91-baae-4228-8828-c59db58dcf36.jpg" width=800 alt="base class">
</div>

View File

@ -509,7 +509,7 @@ model = dict(
### Reuse variables in \_base\_ file
If the users want to reuse the variables in the base file, they can get a copy of the corresponding variable by using `{{_base_.xxx}}`. The latest version of MMEngine also support reusing variables without `{{}}` usage.
If the users want to reuse the variables in the base file, they can get a copy of the corresponding variable by using `{{_base_.xxx}}`. The latest version of MMEngine also support reusing variables without `{{}}` usage.
E.g: