wongkinyiu / crossstagepartialnetworks Goto Github PK
View Code? Open in Web Editor NEWCross Stage Partial Networks
Home Page: https://github.com/WongKinYiu/CrossStagePartialNetworks
Cross Stage Partial Networks
Home Page: https://github.com/WongKinYiu/CrossStagePartialNetworks
Hi,
I am interested in CSPNet recently, and reading the paper: https://arxiv.org/pdf/1911.11929.pdf.
But I have a question about the gradient calculation in page 4, in the paper the gradient calculate as
w1' = f(w1, g0)
w2' = f(w2, g0, g1)
...
wk' = f(wk, g0, g1, g2, ..., gk-1)
Don't this part is calculated as this?
w1' = f(w1, g0, g1, g2, ..., gk)
w2' = f(w2, g1, g2, ..., gk)
...
wk' = f(wk, gk)
also I want to confirm that if the definition of gi is the partial differential of error to weight? that is,
I was very confuse about this part, hope that you can help me.
I am interested in CSPPeleeNet. since you didn't provide it's cfg yet, I planned to do it myself. but according to your cfg files of resnet50 and CSPresnet50, from resnet50 to CSPresnet50, you did not only add the CSP connections but also change other structure of the net, such as the number of the blocks: [3 4 6 3] vs. [3 3 5 2], and also the number of filter. so could you provide the cfg of CSPPeleeNet? Thanks.
你好,我看你的配置文件cfg/csdarknet53.cfg里面decay=0.0005,但是你的cspnet论文里面写的weight_decay=0.005,包括yolov4论文里面也是写的weight_decay=0.005,实际训练cspdarknet53分类这个差别大吗?
@WongKinYiu I have been trying to set right hyper parameters for yolov3-spp but for complete open-images dataset after 300-400 iterations server restarts I have previously trained with 3 of the classes among 601 classes but then I used single GPU parameters for multi GPU training when training for threee classes and dataset size is also small then like 1100 images or So. But now training on whole dataset with Multi GPU parameters causing system reboot,
BTW how do you calculate the parameters for multi-GPU you have already replied to me in previous issues on @AlexeyAB repo at the core how to set burn-in,learning rate, decay in cfg file.as narrated by alexy is causing issue so I changed almost all the parameters to single GPU config except burn-in even then problem persists
For the above hardware here is the link to config I'm using
Please help me out
Thanks
i see paper EFM part that gree arrows means Fusion.how to do that?
Hi @WongKinYiu i want to ask a question. From pjreddie website : image classification model can be trained and inferenced using the darknet framework https://pjreddie.com/darknet/imagenet/, but is there any way to do inference of the classifier model by not using darknet framework? Like the object detection model can be inferenced by using opencvdnn or by converting it to another framework ? Can you kindly give me advice ? Thankyou so much before.
Hi,
I've been using Darknet (Alexey) for a while and I am familiar with how to configure a cfg for a custom training for object detection. This repository is suggested to me and the claimed results seem promising.
But, I could not find how to use a cfg for a training. For instance, I could not figure out how to adapt the cfg to number of classes I'd like to train for. In regular Darknet, one has to configure some parameters in Yolo layer (number of filters, classes etc) in order to adapt the cfg to a custom training.
Any help is appreciated
Do you have a darknet hardnet implementation?
Hi,
In ImageNet Experiments, the paper said that it should be trained for 800 epochs:
However, in the code, it said that it should be trained for 80 epochs:
So there is a big difference……
Besides, I try to re-implement in PyTorch, and the ACC is 7~8 points behind your method. The network architecture and number of parameters is the same as your Darknet results……
Best,
Mu
Hi,
I've been inspecting CSPResNeXt50-PANet-SPP for human detection in real-time. According to readme file of this repository, CSPResNeXt50-PANet-SPP performs better than Yolov3 in AP on COCO dataset.
In order to verify this result, I downloaded cfg and weights of CSPResNeXt50-PANet-SPP to compare it with Yolov3 (yolov3.cfg + yolov3.weights - result of COCO training).
As far as I could observe, CSPResNeXt50-PANet-SPP is not better than Yolov3 at least for my case of detecting humans in video streams. Here is an example image of results of both networks:
My question is that whether these images represent a special case where CSPResNeXt50-PANet-SPP may perform worse than Yolov3? For instance, maybe for small objects like humans in the given images? Or what is the best way to explain this status?
Thanks in advance.
Hi,
I'd like to train with model CSPResNeXt-50, and I downloaded the csresnext50.cfg from https://github.com/WongKinYiu/CrossStagePartialNetworks#big-models.
I checked the .cfg file with netron(https://netron.app/). It seems that csresnext50.cfg do not split the base layer in the stage as mentioned in paper, is it?
Hi @WongKinYiu
Thanks for your answer to my previous question. However, I found severe FLOPs mismatch in your paper and darknet paper @AlexeyAB .
For example, as for ResNet50, the original paper states that the FLOPs is around 3.8 to 4.0 G FLOPs. However, in your paper, it is 9.74 BFLOPs. So there is a big difference. I found maybe there is a ratio of 2.59 around them, right?
RuntimeError: shape '[512, 2048, 1, 1]' is invalid for input of size 680093
Can you tell me what is wrong?
Hi,
I wanted to know if the cfg and weights for CSPDenseNet-201+Elastic will be made available in the near future?
Great work and Thanks
@WongKinYiu Hi,
Do you know why just one small change (1 line) greatly improves accuracy + 0.7
on Pytorch-yolo? https://github.com/WongKinYiu/CrossStagePartialNetworks/blob/pytorch/README.md
Model | Size | NMS | 1080ti fps | BFLOPs | AP | AP50 | AP75 | cfg | weight |
---|---|---|---|---|---|---|---|---|---|
CSPResNeXt50-PANet-SPP | 512×512 | 0.5 | 44 | 71.331 | 39.2 | 59.5 | 41.8 | cfg | - |
CSPResNeXt50c-PANet-SPP | 512×512 | 0.5 | - | 71.734 | 39.9 | 60.1 | 42.6 | cfg | - |
@WongKinYiu Hi,
Did you compare AP
(MS COCO) and 1080ti fps
(rather than BFLOPS) of models ?
CSPResNeXt50-PANet-SPP-GIoU https://github.com/WongKinYiu/CrossStagePartialNetworks#gpu-real-time-models
CSPDarkNet-53-SPP-GIoU-Yolov3 that is based on CSPDarkNet-53
https://github.com/WongKinYiu/CrossStagePartialNetworks#big-models
default Yolov3-SPP model
With the same: network resolution, mini_batch=batch/subdivisions, data augmentation, the same dataset - to compare apples with apples.
In the preprint paper, I saw the difference result when test in the MSCOCO object detection task. For instance, with size 608x608 CSPResNeXt50-PANet-SPP had AP is 43.2 in the repos comparing with 38.4 in the paper. It's a big gapped. Could you explain me clearly?
Capture from ReadMe.md:
From preprint paper:
One more question, the input size of csresnext50-panet-spp-original-optimal.cfg
is 512, the last yolo output sequently is 64,32,16
so when you getting optimal model on input size 608 it will 76,38,19
, does you need modify model to get the best result?
I want to retrain csmobilenet-v2.cfg for a different task starting with the provided weights.
How many layers should I extract from the provided weights?
There are 59 [convolutional] layers and since other layers like max-pooling and routes don't have weights, and since I am changing the number of convolutional filters immediately before the softmax, I guess that the answer is 58?
Like:
./darknet partial csmobilenet-v2.cfg csmobilenet-v2.weights csmobilenet-v2.conv.58 58
Hi, thanks for your work and sharing.
I want to know why can you add so many 1x1 filters and copy operations, but parameters and BFLOPs reduced.
@WongKinYiu @AlexeyAB great work on the README, there's a wealth of information here! I saw that the mosaic augmentation outperformed all of the rest in your readme tests, including more well known ones like cutmix.
Based on this I wonder if it's worth publishing a short paper to arxiv, and then we can link to this github, alexeyab/darknet and ultralytics/yolov3 for the mosaic dataloader code?
Hi, I'm writing an article and have found/tested your implementation of the following structures:
CSPResNeXt50-PANet-SPP
CSPDarknet53-PANet-SPP
CSPResNet50-PANet-SPP
The question is do you have an article which I can cite referring your implementation?
Great Work!!
Can you provide this work in keras?
Hi,
I need to train and test with csmobilenet-v2.cfg on coco, how to train it ???
Thanks
I am trying to combine csp into mobilenet pytorch. But hard to read the darknet config of that model. Can you visualize this model in somes tool like netron?
Hello! Thanks for your great work! I want to use "csdarknet53s-panet-spp" and "yolov3-spp" in my master thesis. I understand yolov3-spp however I don't quite understand csdarknet53s-panet-spp.
Also for training csdarknet53s-panet-spp we need "csdarknet53.conv.104" as you wrote in issue #17. However I can't find the file. Could you please link it?
@WongKinYiu Hi,
Since CSPDarkNet53 is better than CSPResNeXt50 for Detector, try to train these 4 models:
Model | GPU | 256x256 | 512x512 | 608x608 |
---|---|---|---|---|
darknet53.cfg (original) | RTX 2070 | 113 | 56 | 38 |
csdarknet53.cfg (original) | RTX 2070 | 101 | 57 | 41 |
csdarknet53g.cfg.txt | RTX 2070 | 122 | 64 | 46 |
csdarknet53ghr.cfg.txt | RTX 2070 | 100 | 75 | 57 |
spinenet49.cfg.txt low priority | RTX 2070 | 49 | 44 | 43 |
csdarknet19-fast.cfg.txt | RTX 2070 | 213 | 149 | 116 |
csdarknet19-fast.cfg
contains DropBlock, so use the latest version of Darknet that uses fast random-functions for DropBlock.
I have a flower dataset with min size of 10x10 objects.
Is there any cfg files for SNet146-ThunderNet to train ?
@WongKinYiu Hi,
I have a problem that sometimes some pictures are not detected or detected wrong. I attached my model and some images for testing. Could you please check it and guide me? I have about 2K images per class. Please give me some information about the hyperparameters for my case.
Thanks in advance
Originally posted by @zpmmehrdad in #6 (comment)
Due to the limitation of GPU devices, I only tested the model with epoch = 1, and found that compared with the traditional resnext model, the result of cspresnext model for an epoch is not satisfactory. Is it because of the residual link used that the model needs more time to learn
你好:
我看了你的csdarknet53.cfg,里面有如下配置:
angle=7
hue=.1
saturation=.75
exposure=.75
aspect=.75
所以这些增强在imagenet数据集上训练时也加进去了,才得到top1 acc:77%的结果吗?
Hi @WongKinYiu thanks for the great work here! I introduced new error-checking in https://github.com/ultralytics/yolov3 for custom cfgs, and after running your yolov3-spp-matrix.cfg through, I just realized my implementation is not handling your 'share_index' field correctly. What do I do exactly with the value share_index=115
here?
AssertionError: Unsupported field 'share_index' in cfg/yolov3-spp-matrix.cfg.
CrossStagePartialNetworks/in progress/yolov3-spp-matrix.cfg
Lines 926 to 933 in ff762e5
Chien-Yao Wang,您好!我是Amusi,我想了解壹下妳測試關於YOLOv4和YOLOv5的性能比較結果。因為我今天剛看到妳在妳CSPNet開源項目中提交的issue,裏面還@了我#32
但我當時沒有及時看到,所以今天才看到這個已關閉的issue。
不知道您有沒有微信(wechat)或者知乎也性,能即時通訊的應用即可。我想快速跟您了解壹下您測試的YOLOv4和YOLOv5!
如有打擾,還請諒解
Hi, Can you help me with the selection of the pre-trained weights. I am trying to train the model for custom object training.
how to split the feature in two
in your paper
i see, the frist feture map was splitted in two, one for concat, and another for conv and copy.
But I didn't find any relevant information in your train cfg file: csresnext50-panet-spp-original-optimal.cfg
@WongKinYiu @AlexeyAB
I use CSPDarknet as the backbone of my detection on mmdetection, but the inference time of CSPDarknet is slower than Darknet when batchsize is 1. However, when batchsize increases 8 or more, the inference time is quicker than Darknet.
Is this normal? or something wrong?
Thanks!
What's mean CIO?I can't understand, please tell me the full name of CIO!
I used the csresnext50-panet-spp-original-optimal.cfg and the weight file associated with it. I followed the steps mentioned by @AlexeyAB's yolo repo and changed my cfg accordingly.
I changed all convolution filters before yolo to 18 as I have 1 class and reduced the max_batches to 10000. I also set the steps to 8000,9000.
But I can never start the training. It loads the model and just stops.
How do I fix this?
in the cfg file, the first shortcut layer cut is from -4, which is convolutional layer with filters=64, while the convolutional layer before this [shortcut] is filters=128, they cannot be added, is this an error?
[net]
#batch=1
#subdivisions=1
batch=64
subdivisions=8
width=512
height=512
channels=3
momentum=0.949
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1
learning_rate=0.00261
burn_in=1000
max_batches = 500500
policy=steps
steps=400000,450000
scales=.1,.1
#cutmix=1
mosaic=1
#19:104x104 38:52x52 65:26x26 80:13x13 for 416
[convolutional]
batch_normalize=1
filters=64
size=7
stride=2
pad=1
activation=leaky
[maxpool]
size=2
stride=2
[convolutional]
batch_normalize=1
filters=128
size=1
stride=1
pad=1
activation=leaky
[route]
layers = -2
[convolutional]
batch_normalize=1
filters=64
size=1
stride=1
pad=1
activation=leaky
[convolutional]
batch_normalize=1
filters=128
size=1
stride=1
pad=1
activation=leaky
[convolutional]
batch_normalize=1
filters=128
size=3
groups=32
stride=1
pad=1
activation=leaky
[convolutional]
batch_normalize=1
filters=128
size=1
stride=1
pad=1
activation=linear
[shortcut]
from=-4
activation=leaky
I read the paper that you train the model in single GPU. How long does it take to train on a single GPU? Why not train on multi-GPU?
I'm reading the article for CSPNet: A New Backbone that can Enhance Learning Capability of CNN.
It is not quite clear Equation (6) shown the image below:
Ho did you derive the equation (6)?
Also it's not clear the definition of g_i
represents the gradient propagated to the i^th dense layer. Could you explain more detail?
Thank you.
I read through the repository but can't figure out which model is used in that section. Could you please tell me where I can find it? Thank you.
I used the csdarknet53-omega. cfg file to train the classification network(not detection) on my own dataset. When using label_smooth_eps =0.1, the loss increases gradually, no matter how much learning rate I set. But when not use label_smooth_eps=0.1, the loss gradually converges.I mean, why is that?
By the way, I have five categories with an average of 1,800 pictures in each category. How many training times and batch sizes should be set?
hello,
I am trying to build a new small & efficient model by concatenating (backbone) csmobilenetv2 + (head) yolov3_tiny to detect objects. Everything going well until ~ 1000 iters (avg loss ~ 8.0) but it gets a nan after that.
Does anyone have any ideas? Thank you.
here is my layer information.
batch = 1, time_steps = 1, train = 0
layer filters size/strd(dil) input output
0 conv 32 3 x 3/ 2 224 x 224 x 3 -> 112 x 112 x 32 0.022 BF
1 conv 16/ 16 3 x 3/ 1 112 x 112 x 32 -> 112 x 112 x 16 0.007 BF
2 route 0 -> 112 x 112 x 32
3 conv 32 1 x 1/ 1 112 x 112 x 32 -> 112 x 112 x 32 0.026 BF
4 conv 32/ 32 3 x 3/ 1 112 x 112 x 32 -> 112 x 112 x 32 0.007 BF
5 conv 16 1 x 1/ 1 112 x 112 x 32 -> 112 x 112 x 16 0.013 BF
6 route 5 1 -> 112 x 112 x 32
7 conv 48 1 x 1/ 1 112 x 112 x 32 -> 112 x 112 x 48 0.039 BF
8 conv 48/ 48 3 x 3/ 2 112 x 112 x 48 -> 56 x 56 x 48 0.003 BF
9 route 7 -> 112 x 112 x 48
10 conv 96/ 48 3 x 3/ 2 112 x 112 x 48 -> 56 x 56 x 96 0.005 BF
11 conv 24 1 x 1/ 1 56 x 56 x 96 -> 56 x 56 x 24 0.014 BF
12 conv 72 1 x 1/ 1 56 x 56 x 24 -> 56 x 56 x 72 0.011 BF
13 conv 144/ 72 3 x 3/ 1 56 x 56 x 72 -> 56 x 56 x 144 0.008 BF
14 conv 24 1 x 1/ 1 56 x 56 x 144 -> 56 x 56 x 24 0.022 BF
15 Shortcut Layer: 11, wt = 0, wn = 0, outputs: 56 x 56 x 24 0.000 BF
16 route 15 8 -> 56 x 56 x 72
17 conv 72 1 x 1/ 1 56 x 56 x 72 -> 56 x 56 x 72 0.033 BF
18 conv 72/ 72 3 x 3/ 2 56 x 56 x 72 -> 28 x 28 x 72 0.001 BF
19 route 17 -> 56 x 56 x 72
20 conv 144/ 72 3 x 3/ 2 56 x 56 x 72 -> 28 x 28 x 144 0.002 BF
21 conv 32 1 x 1/ 1 28 x 28 x 144 -> 28 x 28 x 32 0.007 BF
22 conv 96 1 x 1/ 1 28 x 28 x 32 -> 28 x 28 x 96 0.005 BF
23 conv 192/ 96 3 x 3/ 1 28 x 28 x 96 -> 28 x 28 x 192 0.003 BF
24 conv 32 1 x 1/ 1 28 x 28 x 192 -> 28 x 28 x 32 0.010 BF
25 Shortcut Layer: 21, wt = 0, wn = 0, outputs: 28 x 28 x 32 0.000 BF
26 conv 96 1 x 1/ 1 28 x 28 x 32 -> 28 x 28 x 96 0.005 BF
27 conv 192/ 96 3 x 3/ 1 28 x 28 x 96 -> 28 x 28 x 192 0.003 BF
28 conv 32 1 x 1/ 1 28 x 28 x 192 -> 28 x 28 x 32 0.010 BF
29 Shortcut Layer: 25, wt = 0, wn = 0, outputs: 28 x 28 x 32 0.000 BF
30 conv 96 1 x 1/ 1 28 x 28 x 32 -> 28 x 28 x 96 0.005 BF
31 conv 192/ 96 3 x 3/ 1 28 x 28 x 96 -> 28 x 28 x 192 0.003 BF
32 conv 64 1 x 1/ 1 28 x 28 x 192 -> 28 x 28 x 64 0.019 BF
33 conv 192 1 x 1/ 1 28 x 28 x 64 -> 28 x 28 x 192 0.019 BF
34 conv 384/ 192 3 x 3/ 1 28 x 28 x 192 -> 28 x 28 x 384 0.005 BF
35 conv 64 1 x 1/ 1 28 x 28 x 384 -> 28 x 28 x 64 0.039 BF
36 Shortcut Layer: 32, wt = 0, wn = 0, outputs: 28 x 28 x 64 0.000 BF
37 conv 192 1 x 1/ 1 28 x 28 x 64 -> 28 x 28 x 192 0.019 BF
38 conv 384/ 192 3 x 3/ 1 28 x 28 x 192 -> 28 x 28 x 384 0.005 BF
39 conv 64 1 x 1/ 1 28 x 28 x 384 -> 28 x 28 x 64 0.039 BF
40 Shortcut Layer: 36, wt = 0, wn = 0, outputs: 28 x 28 x 64 0.000 BF
41 conv 192 1 x 1/ 1 28 x 28 x 64 -> 28 x 28 x 192 0.019 BF
42 conv 384/ 192 3 x 3/ 1 28 x 28 x 192 -> 28 x 28 x 384 0.005 BF
43 conv 64 1 x 1/ 1 28 x 28 x 384 -> 28 x 28 x 64 0.039 BF
44 Shortcut Layer: 40, wt = 0, wn = 0, outputs: 28 x 28 x 64 0.000 BF
45 route 44 18 -> 28 x 28 x 136
46 conv 192 1 x 1/ 1 28 x 28 x 136 -> 28 x 28 x 192 0.041 BF
47 conv 192/ 192 3 x 3/ 2 28 x 28 x 192 -> 14 x 14 x 192 0.001 BF
48 route 46 -> 28 x 28 x 192
49 conv 384/ 192 3 x 3/ 2 28 x 28 x 192 -> 14 x 14 x 384 0.001 BF
50 conv 96 1 x 1/ 1 14 x 14 x 384 -> 14 x 14 x 96 0.014 BF
51 conv 288 1 x 1/ 1 14 x 14 x 96 -> 14 x 14 x 288 0.011 BF
52 conv 576/ 288 3 x 3/ 1 14 x 14 x 288 -> 14 x 14 x 576 0.002 BF
53 conv 96 1 x 1/ 1 14 x 14 x 576 -> 14 x 14 x 96 0.022 BF
54 Shortcut Layer: 50, wt = 0, wn = 0, outputs: 14 x 14 x 96 0.000 BF
55 conv 288 1 x 1/ 1 14 x 14 x 96 -> 14 x 14 x 288 0.011 BF
56 conv 576/ 288 3 x 3/ 1 14 x 14 x 288 -> 14 x 14 x 576 0.002 BF
57 conv 96 1 x 1/ 1 14 x 14 x 576 -> 14 x 14 x 96 0.022 BF
58 Shortcut Layer: 54, wt = 0, wn = 0, outputs: 14 x 14 x 96 0.000 BF
59 route 58 47 -> 14 x 14 x 288
60 conv 288 1 x 1/ 1 14 x 14 x 288 -> 14 x 14 x 288 0.033 BF
61 conv 288/ 288 3 x 3/ 2 14 x 14 x 288 -> 7 x 7 x 288 0.000 BF
62 route 60 -> 14 x 14 x 288
63 conv 576/ 288 3 x 3/ 2 14 x 14 x 288 -> 7 x 7 x 576 0.001 BF
64 conv 160 1 x 1/ 1 7 x 7 x 576 -> 7 x 7 x 160 0.009 BF
65 conv 480 1 x 1/ 1 7 x 7 x 160 -> 7 x 7 x 480 0.008 BF
66 conv 960/ 480 3 x 3/ 1 7 x 7 x 480 -> 7 x 7 x 960 0.001 BF
67 conv 160 1 x 1/ 1 7 x 7 x 960 -> 7 x 7 x 160 0.015 BF
68 Shortcut Layer: 64, wt = 0, wn = 0, outputs: 7 x 7 x 160 0.000 BF
69 conv 480 1 x 1/ 1 7 x 7 x 160 -> 7 x 7 x 480 0.008 BF
70 conv 960/ 480 3 x 3/ 1 7 x 7 x 480 -> 7 x 7 x 960 0.001 BF
71 conv 160 1 x 1/ 1 7 x 7 x 960 -> 7 x 7 x 160 0.015 BF
72 Shortcut Layer: 68, wt = 0, wn = 0, outputs: 7 x 7 x 160 0.000 BF
73 conv 480 1 x 1/ 1 7 x 7 x 160 -> 7 x 7 x 480 0.008 BF
74 conv 960/ 480 3 x 3/ 1 7 x 7 x 480 -> 7 x 7 x 960 0.001 BF
75 conv 320 1 x 1/ 1 7 x 7 x 960 -> 7 x 7 x 320 0.030 BF
76 route 75 61 -> 7 x 7 x 608
77 conv 640 1 x 1/ 1 7 x 7 x 608 -> 7 x 7 x 640 0.038 BF
78 conv 27 1 x 1/ 1 7 x 7 x 640 -> 7 x 7 x 27 0.002 BF
79 yolo
[yolo] params: iou loss: mse (2), iou_norm: 0.75, cls_norm: 1.00, scale_x_y: 1.00
80 route 75 -> 7 x 7 x 320
81 conv 128 1 x 1/ 1 7 x 7 x 320 -> 7 x 7 x 128 0.004 BF
82 upsample 2x 7 x 7 x 128 -> 14 x 14 x 128
83 route 82 60 -> 14 x 14 x 416
84 conv 256 3 x 3/ 1 14 x 14 x 416 -> 14 x 14 x 256 0.376 BF
85 conv 27 1 x 1/ 1 14 x 14 x 256 -> 14 x 14 x 27 0.003 BF
86 yolo
Can anyone help me using pretrained weight and config to train and inference classification problem?
I don't know any way to apply pretrained weight and config.
Thank you very much.
I have trained my own model csmobilenetv2(backbone) and yolov3_tiny(head) and got the best final weights. Now I want to accelerate the inference with TensorRT which verision is 5.1.6.1. But got errors when translating the weights model to onnx model. Useing yolov3.weights yolov3.cfg yolov3-tiny.weights yolov3-tiny.cfg are ok.
tensorrt:5.1.6.1
python:3.6.9
onnx:1.4.1
numpy:1.18.4
Traceback (most recent call last):
File "yolov3_to_onnx.py", line 840, in
main()
File "yolov3_to_onnx.py", line 827, in main
verbose=True)
File "yolov3_to_onnx.py", line 447, in build_onnx_graph
params)
File "yolov3_to_onnx.py", line 322, in load_conv_weights
conv_params, 'conv', 'weights')
File "yolov3_to_onnx.py", line 351, in _create_param_tensors
conv_params, param_category, suffix)
File "yolov3_to_onnx.py", line 383, in _load_one_param_type
buffer=self.weights_file.read(param_size * 4))
TypeError: buffer is too small for requested array
Excuse me,I noticed that in issue #32 , AlexeyAB and you talked about P6 model, what model does it represent? I don't see it in both CSPNet and YOLOv4 paper.
A declarative, efficient, and flexible JavaScript library for building user interfaces.
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google ❤️ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.