python调用yolov3检测_使用pytorch实现目标检测_YOLOV3实现

objdet_darknet

使用pytorch实现YOLO(v3)。

参考资料

TODO

使用人脸数据集实现人脸检测(face detection)。

网络结构

----------------------------------------------------------------

Layer (type) Output Shape Param #

================================================================

Conv2d-1 [-1, 32, 416, 416] 864

BatchNorm2d-2 [-1, 32, 416, 416] 64

LeakyReLU-3 [-1, 32, 416, 416] 0

Conv2d-4 [-1, 64, 208, 208] 18,432

BatchNorm2d-5 [-1, 64, 208, 208] 128

LeakyReLU-6 [-1, 64, 208, 208] 0

Conv2d-7 [-1, 32, 208, 208] 2,048

BatchNorm2d-8 [-1, 32, 208, 208] 64

LeakyReLU-9 [-1, 32, 208, 208] 0

Conv2d-10 [-1, 64, 208, 208] 18,432

BatchNorm2d-11 [-1, 64, 208, 208] 128

LeakyReLU-12 [-1, 64, 208, 208] 0

Conv2d-13 [-1, 128, 104, 104] 73,728

BatchNorm2d-14 [-1, 128, 104, 104] 256

LeakyReLU-15 [-1, 128, 104, 104] 0

Conv2d-16 [-1, 64, 104, 104] 8,192

BatchNorm2d-17 [-1, 64, 104, 104] 128

LeakyReLU-18 [-1, 64, 104, 104] 0

Conv2d-19 [-1, 128, 104, 104] 73,728

BatchNorm2d-20 [-1, 128, 104, 104] 256

LeakyReLU-21 [-1, 128, 104, 104] 0

Conv2d-22 [-1, 64, 104, 104] 8,192

BatchNorm2d-23 [-1, 64, 104, 104] 128

LeakyReLU-24 [-1, 64, 104, 104] 0

Conv2d-25 [-1, 128, 104, 104] 73,728

BatchNorm2d-26 [-1, 128, 104, 104] 256

LeakyReLU-27 [-1, 128, 104, 104] 0

Conv2d-28 [-1, 256, 52, 52] 294,912

BatchNorm2d-29 [-1, 256, 52, 52] 512

LeakyReLU-30 [-1, 256, 52, 52] 0

Conv2d-31 [-1, 128, 52, 52] 32,768

BatchNorm2d-32 [-1, 128, 52, 52] 256

LeakyReLU-33 [-1, 128, 52, 52] 0

Conv2d-34 [-1, 256, 52, 52] 294,912

BatchNorm2d-35 [-1, 256, 52, 52] 512

LeakyReLU-36 [-1, 256, 52, 52] 0

Conv2d-37 [-1, 128, 52, 52] 32,768

BatchNorm2d-38 [-1, 128, 52, 52] 256

LeakyReLU-39 [-1, 128, 52, 52] 0

Conv2d-40 [-1, 256, 52, 52] 294,912

BatchNorm2d-41 [-1, 256, 52, 52] 512

LeakyReLU-42 [-1, 256, 52, 52] 0

Conv2d-43 [-1, 128, 52, 52] 32,768

BatchNorm2d-44 [-1, 128, 52, 52] 256

LeakyReLU-45 [-1, 128, 52, 52] 0

Conv2d-46 [-1, 256, 52, 52] 294,912

BatchNorm2d-47 [-1, 256, 52, 52] 512

LeakyReLU-48 [-1, 256, 52, 52] 0

Conv2d-49 [-1, 128, 52, 52] 32,768

BatchNorm2d-50 [-1, 128, 52, 52] 256

LeakyReLU-51 [-1, 128, 52, 52] 0

Conv2d-52 [-1, 256, 52, 52] 294,912

BatchNorm2d-53 [-1, 256, 52, 52] 512

LeakyReLU-54 [-1, 256, 52, 52] 0

Conv2d-55 [-1, 128, 52, 52] 32,768

BatchNorm2d-56 [-1, 128, 52, 52] 256

LeakyReLU-57 [-1, 128, 52, 52] 0

Conv2d-58 [-1, 256, 52, 52] 294,912

BatchNorm2d-59 [-1, 256, 52, 52] 512

LeakyReLU-60 [-1, 256, 52, 52] 0

Conv2d-61 [-1, 128, 52, 52] 32,768

BatchNorm2d-62 [-1, 128, 52, 52] 256

LeakyReLU-63 [-1, 128, 52, 52] 0

Conv2d-64 [-1, 256, 52, 52] 294,912

BatchNorm2d-65 [-1, 256, 52, 52] 512

LeakyReLU-66 [-1, 256, 52, 52] 0

Conv2d-67 [-1, 128, 52, 52] 32,768

BatchNorm2d-68 [-1, 128, 52, 52] 256

LeakyReLU-69 [-1, 128, 52, 52] 0

Conv2d-70 [-1, 256, 52, 52] 294,912

BatchNorm2d-71 [-1, 256, 52, 52] 512

LeakyReLU-72 [-1, 256, 52, 52] 0

Conv2d-73 [-1, 128, 52, 52] 32,768

BatchNorm2d-74 [-1, 128, 52, 52] 256

LeakyReLU-75 [-1, 128, 52, 52] 0

Conv2d-76 [-1, 256, 52, 52] 294,912

BatchNorm2d-77 [-1, 256, 52, 52] 512

LeakyReLU-78 [-1, 256, 52, 52] 0

Conv2d-79 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-80 [-1, 512, 26, 26] 1,024

LeakyReLU-81 [-1, 512, 26, 26] 0

Conv2d-82 [-1, 256, 26, 26] 131,072

BatchNorm2d-83 [-1, 256, 26, 26] 512

LeakyReLU-84 [-1, 256, 26, 26] 0

Conv2d-85 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-86 [-1, 512, 26, 26] 1,024

LeakyReLU-87 [-1, 512, 26, 26] 0

Conv2d-88 [-1, 256, 26, 26] 131,072

BatchNorm2d-89 [-1, 256, 26, 26] 512

LeakyReLU-90 [-1, 256, 26, 26] 0

Conv2d-91 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-92 [-1, 512, 26, 26] 1,024

LeakyReLU-93 [-1, 512, 26, 26] 0

Conv2d-94 [-1, 256, 26, 26] 131,072

BatchNorm2d-95 [-1, 256, 26, 26] 512

LeakyReLU-96 [-1, 256, 26, 26] 0

Conv2d-97 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-98 [-1, 512, 26, 26] 1,024

LeakyReLU-99 [-1, 512, 26, 26] 0

Conv2d-100 [-1, 256, 26, 26] 131,072

BatchNorm2d-101 [-1, 256, 26, 26] 512

LeakyReLU-102 [-1, 256, 26, 26] 0

Conv2d-103 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-104 [-1, 512, 26, 26] 1,024

LeakyReLU-105 [-1, 512, 26, 26] 0

Conv2d-106 [-1, 256, 26, 26] 131,072

BatchNorm2d-107 [-1, 256, 26, 26] 512

LeakyReLU-108 [-1, 256, 26, 26] 0

Conv2d-109 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-110 [-1, 512, 26, 26] 1,024

LeakyReLU-111 [-1, 512, 26, 26] 0

Conv2d-112 [-1, 256, 26, 26] 131,072

BatchNorm2d-113 [-1, 256, 26, 26] 512

LeakyReLU-114 [-1, 256, 26, 26] 0

Conv2d-115 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-116 [-1, 512, 26, 26] 1,024

LeakyReLU-117 [-1, 512, 26, 26] 0

Conv2d-118 [-1, 256, 26, 26] 131,072

BatchNorm2d-119 [-1, 256, 26, 26] 512

LeakyReLU-120 [-1, 256, 26, 26] 0

Conv2d-121 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-122 [-1, 512, 26, 26] 1,024

LeakyReLU-123 [-1, 512, 26, 26] 0

Conv2d-124 [-1, 256, 26, 26] 131,072

BatchNorm2d-125 [-1, 256, 26, 26] 512

LeakyReLU-126 [-1, 256, 26, 26] 0

Conv2d-127 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-128 [-1, 512, 26, 26] 1,024

LeakyReLU-129 [-1, 512, 26, 26] 0

Conv2d-130 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-131 [-1, 1024, 13, 13] 2,048

LeakyReLU-132 [-1, 1024, 13, 13] 0

Conv2d-133 [-1, 512, 13, 13] 524,288

BatchNorm2d-134 [-1, 512, 13, 13] 1,024

LeakyReLU-135 [-1, 512, 13, 13] 0

Conv2d-136 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-137 [-1, 1024, 13, 13] 2,048

LeakyReLU-138 [-1, 1024, 13, 13] 0

Conv2d-139 [-1, 512, 13, 13] 524,288

BatchNorm2d-140 [-1, 512, 13, 13] 1,024

LeakyReLU-141 [-1, 512, 13, 13] 0

Conv2d-142 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-143 [-1, 1024, 13, 13] 2,048

LeakyReLU-144 [-1, 1024, 13, 13] 0

Conv2d-145 [-1, 512, 13, 13] 524,288

BatchNorm2d-146 [-1, 512, 13, 13] 1,024

LeakyReLU-147 [-1, 512, 13, 13] 0

Conv2d-148 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-149 [-1, 1024, 13, 13] 2,048

LeakyReLU-150 [-1, 1024, 13, 13] 0

Conv2d-151 [-1, 512, 13, 13] 524,288

BatchNorm2d-152 [-1, 512, 13, 13] 1,024

LeakyReLU-153 [-1, 512, 13, 13] 0

Conv2d-154 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-155 [-1, 1024, 13, 13] 2,048

LeakyReLU-156 [-1, 1024, 13, 13] 0

Conv2d-157 [-1, 512, 13, 13] 524,288

BatchNorm2d-158 [-1, 512, 13, 13] 1,024

LeakyReLU-159 [-1, 512, 13, 13] 0

Conv2d-160 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-161 [-1, 1024, 13, 13] 2,048

LeakyReLU-162 [-1, 1024, 13, 13] 0

Conv2d-163 [-1, 512, 13, 13] 524,288

BatchNorm2d-164 [-1, 512, 13, 13] 1,024

LeakyReLU-165 [-1, 512, 13, 13] 0

Conv2d-166 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-167 [-1, 1024, 13, 13] 2,048

LeakyReLU-168 [-1, 1024, 13, 13] 0

Conv2d-169 [-1, 512, 13, 13] 524,288

BatchNorm2d-170 [-1, 512, 13, 13] 1,024

LeakyReLU-171 [-1, 512, 13, 13] 0

Conv2d-172 [-1, 1024, 13, 13] 4,718,592

BatchNorm2d-173 [-1, 1024, 13, 13] 2,048

LeakyReLU-174 [-1, 1024, 13, 13] 0

Conv2d-175 [-1, 255, 13, 13] 261,375

Conv2d-176 [-1, 256, 13, 13] 131,072

BatchNorm2d-177 [-1, 256, 13, 13] 512

LeakyReLU-178 [-1, 256, 13, 13] 0

Upsample-179 [-1, 256, 26, 26] 0

Conv2d-180 [-1, 256, 26, 26] 196,608

BatchNorm2d-181 [-1, 256, 26, 26] 512

LeakyReLU-182 [-1, 256, 26, 26] 0

Conv2d-183 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-184 [-1, 512, 26, 26] 1,024

LeakyReLU-185 [-1, 512, 26, 26] 0

Conv2d-186 [-1, 256, 26, 26] 131,072

BatchNorm2d-187 [-1, 256, 26, 26] 512

LeakyReLU-188 [-1, 256, 26, 26] 0

Conv2d-189 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-190 [-1, 512, 26, 26] 1,024

LeakyReLU-191 [-1, 512, 26, 26] 0

Conv2d-192 [-1, 256, 26, 26] 131,072

BatchNorm2d-193 [-1, 256, 26, 26] 512

LeakyReLU-194 [-1, 256, 26, 26] 0

Conv2d-195 [-1, 512, 26, 26] 1,179,648

BatchNorm2d-196 [-1, 512, 26, 26] 1,024

LeakyReLU-197 [-1, 512, 26, 26] 0

Conv2d-198 [-1, 255, 26, 26] 130,815

Conv2d-199 [-1, 128, 26, 26] 32,768

BatchNorm2d-200 [-1, 128, 26, 26] 256

LeakyReLU-201 [-1, 128, 26, 26] 0

Upsample-202 [-1, 128, 52, 52] 0

Conv2d-203 [-1, 128, 52, 52] 49,152

BatchNorm2d-204 [-1, 128, 52, 52] 256

LeakyReLU-205 [-1, 128, 52, 52] 0

Conv2d-206 [-1, 256, 52, 52] 294,912

BatchNorm2d-207 [-1, 256, 52, 52] 512

LeakyReLU-208 [-1, 256, 52, 52] 0

Conv2d-209 [-1, 128, 52, 52] 32,768

BatchNorm2d-210 [-1, 128, 52, 52] 256

LeakyReLU-211 [-1, 128, 52, 52] 0

Conv2d-212 [-1, 256, 52, 52] 294,912

BatchNorm2d-213 [-1, 256, 52, 52] 512

LeakyReLU-214 [-1, 256, 52, 52] 0

Conv2d-215 [-1, 128, 52, 52] 32,768

BatchNorm2d-216 [-1, 128, 52, 52] 256

LeakyReLU-217 [-1, 128, 52, 52] 0

Conv2d-218 [-1, 256, 52, 52] 294,912

BatchNorm2d-219 [-1, 256, 52, 52] 512

LeakyReLU-220 [-1, 256, 52, 52] 0

Conv2d-221 [-1, 255, 52, 52] 65,535

================================================================

Total params: 61,949,149

Trainable params: 61,949,149

Non-trainable params: 0

----------------------------------------------------------------

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值