Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add：swift yolo and delete other yolo #156

Merged
merged 35 commits into from
Dec 25, 2023
Merged

add：swift yolo and delete other yolo #156

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
35 commits
Select commit Hold shift + click to select a range
7bccb4a
modify: yolov8 inference result foramt
mjq2020 Oct 16, 2023
0ee1513
delete: vscode config
mjq2020 Oct 16, 2023
0dd455f
add: semidataset
mjq2020 Oct 20, 2023
1c9440e
add: dataset sampler for semi
mjq2020 Oct 20, 2023
bfd4277
add: semisup train eval loop and hook
mjq2020 Nov 8, 2023
f1f7098
add: semisup sampler
mjq2020 Nov 8, 2023
c223e5f
add: semisup loss
mjq2020 Nov 8, 2023
bd80826
optimizer: fomo inference
mjq2020 Nov 8, 2023
1a9c232
Add the fu:nction of exporting vela model
mjq2020 Nov 9, 2023
9e21986
Add: the function of installing and checking third-party libraries
mjq2020 Nov 9, 2023
95525b6
modify: sample size is zero bug
mjq2020 Nov 16, 2023
6d81510
add: labelmatch hook
mjq2020 Nov 19, 2023
65fc5f9
optimize: ssod dataset sampler
mjq2020 Nov 19, 2023
3258474
fix: CI error
mjq2020 Nov 19, 2023
01e7055
add: ssod base class
mjq2020 Nov 19, 2023
fefa8b0
add: project init file
mjq2020 Nov 19, 2023
44d961e
add: labelmatch model config
mjq2020 Nov 19, 2023
12b490e
add: unsup dataset class
mjq2020 Nov 19, 2023
437a440
add: EfficientTeacher class
mjq2020 Nov 19, 2023
bdb9893
add: EfficientTeacher class
mjq2020 Nov 19, 2023
133dddf
modify: unsup student get
mjq2020 Nov 19, 2023
24a3d10
fix: register name is class type bug
mjq2020 Nov 19, 2023
65a9144
add: fair pseudo label creator
mjq2020 Nov 19, 2023
1f61eab
add: faster shuffllenetv2 backbone
mjq2020 Nov 27, 2023
f2f117e
add: swift yolo config for 320x320 size
mjq2020 Nov 27, 2023
8208dc0
delete: yolo except swift yolo
mjq2020 Nov 27, 2023
069c6fe
add: optimized swift yolo
mjq2020 Nov 27, 2023
c55fda7
add: this project to the pythonpath environment variable
mjq2020 Dec 22, 2023
ae8346a
fix: no file exit bug
mjq2020 Dec 22, 2023
a6bca6e
modify: thirty library version limit
mjq2020 Dec 22, 2023
2c48589
fix: mmdet bug
mjq2020 Dec 22, 2023
9ec2d9a
Merge branch 'dev' into dev
mjq2020 Dec 22, 2023
90d0f77
fix: ci error bug
mjq2020 Dec 22, 2023
2a487c0
Merge branch 'dev' of https://github.com/mjq2020/EdgeLab into dev
mjq2020 Dec 22, 2023
8eb4ec0
add: init import
mjq2020 Dec 22, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
182 changes: 0 additions & 182 deletions configs/ssd/bash_arch.py
View file
Open in desktop

This file was deleted.

0 configs/yolov5/base_arch.py → configs/swift_yolo/base_arch.py
View file
Open in desktop
File renamed without changes.
28 changes: 24 additions & 4 deletions configs/yolov5/yolov5_s_1xb16_300e_coco.py → .../swift_yolo/swift_yolo_1xb16_300e_coco.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
# MODEL
num_classes = 71
deepen_factor = 0.33
widen_factor = 0.5
widen_factor = 0.15

# DATA
dataset_type = 'sscma.CustomYOLOv5CocoDataset'
Expand All @@ -17,12 +17,11 @@
data_root = 'https://universe.roboflow.com/ds/z5UOcgxZzD?key=bwx9LQUT0t'
height = 640
width = 640
imgsz = (width, height)
batch = 16
workers = 2
val_batch = batch
val_workers = workers

imgsz = (width, height)

# TRAIN
persistent_workers = True
Expand Down Expand Up @@ -86,8 +85,22 @@
dict(type='LoadAnnotations', with_bbox=True),
]

# from mmyolo.datasets.transforms import YOLOv5RandomAffine

color_space = [
[dict(type='mmdet.ColorTransform')],
[dict(type='mmdet.AutoContrast')],
[dict(type='mmdet.Equalize')],
[dict(type='mmdet.Sharpness')],
[dict(type='mmdet.Posterize')],
[dict(type='mmdet.Solarize')],
[dict(type='mmdet.Color')],
[dict(type='mmdet.Contrast')],
[dict(type='mmdet.Brightness')],
]
train_pipeline = [
*pre_transform,
dict(type='LoadImageFromFile', file_client_args=dict(backend='disk')),
dict(type='LoadAnnotations', with_bbox=True),
dict(type='Mosaic', img_scale=imgsz, pad_val=114.0, pre_transform=pre_transform),
dict(
type='YOLOv5RandomAffine',
Expand All @@ -104,6 +117,13 @@
bbox_params=dict(type='BboxParams', format='pascal_voc', label_fields=['gt_bboxes_labels', 'gt_ignore_flags']),
keymap={'img': 'image', 'gt_bboxes': 'bboxes'},
),
dict(
type='mmdet.RandomOrder',
transforms=[
dict(type='mmdet.RandAugment', aug_space=color_space, aug_num=1),
# dict(type='mmdet.RandAugment', aug_space=geometric, aug_num=1),
],
),
dict(type='YOLOv5HSVRandomAug'),
dict(type='mmdet.RandomFlip', prob=0.5),
dict(
Expand Down
Loading