mirror of https://github.com/open-mmlab/mmyolo.git
Fix some config and add description
parent
b59aa0fb5a
commit
c683d1f94d
|
@ -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
|
||||
|
||||
|
|
|
@ -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(
|
||||
|
|
|
@ -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
|
||||
|
||||
|
|
|
@ -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>
|
||||
|
|
|
@ -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:
|
||||
|
||||
|
|
Loading…
Reference in New Issue