YOLO code interpretation_ model.py
1 Overview
This document is mainly composed of
Two classes, Darknet yolayer,Several functions create_modules,get_yolo_layers,load_darknet_weights,save_weights,convert,attempt_download form
Darknet's constructor first calls parse_model_cfg,create_modules,get_yolo_layers are three functions, so let's start with them and then further ...
Posted by aunquarra on Sun, 30 Jan 2022 23:38:35 +0100
Analysis of YOLOV5 network module
preface
[personal learning notes, please correct any errors]
YOLO-V5 code warehouse address: https://github.com/ultralytics/yolov5
1, Conv module
Before introducing each module, you need to introduce the most basic Conv module in YOLOV5.
class Conv(nn.Module):
def __init__(self, c1, c2, k=1, s=1, p=None, g=1, act=True):
# ch_i ...
Posted by monkeymade on Wed, 05 Jan 2022 11:14:30 +0100
YoloV3-SPP NMS source code details
preface
The article is linked to: Detailed analysis of YOLO-V3-SPP This paper mainly explains the processing of YoloV3-SPP in the verification reasoning stage, which is divided into the following points:
Preprocessing dataProcessing of validationReasoning NMS processingdrawbox for visualization of reasoning results
For the calculation of map ...
Posted by reddrum on Thu, 23 Dec 2021 12:07:46 +0100
Reproduce the YOLOv3 data class from scratch
1 data organization
(1) coco128 dataset
Here we use the coco dataset, but the coco dataset is too large. Here we use the coco128 dataset, because there are only 128 pictures, and the speed of copying and decompression can be fast. The download link of the coco128 dataset is: https://github.com/ultralytics/yolov5/releases/download/v1.0/coc ...
Posted by hyabusa on Sun, 31 Oct 2021 09:45:11 +0100