dustysys commited on
Commit
f5c099f
1 Parent(s): 2957cb2

add bbox model

Browse files
mmdet/bbox/mmdet_anime-face_yolov3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38208bb6b8a4633193feba532e96ed9a7942129af8fe948b27bfcf8e9a30a12e
3
+ size 246462357
mmdet/bbox/mmdet_anime-face_yolov3.py ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model = dict(type='YOLOV3',
2
+ backbone=dict(type='Darknet', depth=53, out_indices=(3, 4, 5)),
3
+ neck=dict(type='YOLOV3Neck',
4
+ num_scales=3,
5
+ in_channels=[1024, 512, 256],
6
+ out_channels=[512, 256, 128]),
7
+ bbox_head=dict(type='YOLOV3Head',
8
+ num_classes=1,
9
+ in_channels=[512, 256, 128],
10
+ out_channels=[1024, 512, 256],
11
+ anchor_generator=dict(type='YOLOAnchorGenerator',
12
+ base_sizes=[[(116, 90),
13
+ (156, 198),
14
+ (373, 326)],
15
+ [(30, 61),
16
+ (62, 45),
17
+ (59, 119)],
18
+ [(10, 13),
19
+ (16, 30),
20
+ (33, 23)]],
21
+ strides=[32, 16, 8]),
22
+ bbox_coder=dict(type='YOLOBBoxCoder'),
23
+ featmap_strides=[32, 16, 8]),
24
+ test_cfg=dict(nms_pre=1000,
25
+ min_bbox_size=0,
26
+ score_thr=0.05,
27
+ conf_thr=0.005,
28
+ nms=dict(type='nms', iou_threshold=0.45),
29
+ max_per_img=100))
30
+ test_pipeline = [
31
+ dict(type='LoadImageFromFile'),
32
+ dict(type='MultiScaleFlipAug',
33
+ img_scale=(608, 608),
34
+ flip=False,
35
+ transforms=[
36
+ dict(type='Resize', keep_ratio=True),
37
+ dict(type='RandomFlip'),
38
+ dict(type='Normalize',
39
+ mean=[0, 0, 0],
40
+ std=[255.0, 255.0, 255.0],
41
+ to_rgb=True),
42
+ dict(type='Pad', size_divisor=32),
43
+ dict(type='ImageToTensor', keys=['img']),
44
+ dict(type='Collect', keys=['img'])
45
+ ])
46
+ ]
47
+ data = dict(test=dict(pipeline=test_pipeline))