Coder Social home page Coder Social logo

wujixiu / helmet-detection Goto Github PK

View Code? Open in Web Editor NEW
349.0 25.0 116.0 35.03 MB

Helmet Detection on Construction Sites

License: Apache License 2.0

Python 10.39% C++ 79.19% CMake 2.41% Makefile 0.59% Shell 0.35% Dockerfile 0.06% HTML 0.05% CSS 0.21% Cuda 6.00% MATLAB 0.74%
helmet-detection ssd-pelee deep-learning

helmet-detection's People

Contributors

wujixiu avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

helmet-detection's Issues

询问论文中的fast r cnn

是复现了对比的论文中的fast rcnn么,用自己的数据集重新训练的?能不能提供一下fast rcnn 的预训练模型,我想试试

Labeled region

Labeled region is the head but not the helmet.
It's a trick? raise the mAP?
If labeled region includes only the helmet, would the model converge faster, or not?

Duplicates in the train and test sets

I found that some images are duplicates but have different names. Some of them appear both in test and train set and some others appear either two times in the train set or two times in the test set.

The images names are the following:

./00376.jpg
./00444.jpg

./00239.jpg
./00676.jpg

./00177.jpg
./02078.jpg

./00286.jpg
./02142.jpg

./00292.jpg
./02542.jpg

./02423.jpg
./03286.jpg

./02224.jpg
./03365.jpg

./00615.jpg
./00641.jpg

./00801.jpg
./01007.jpg

./00509.jpg
./00832.jpg

./00545.jpg
./01132.jpg

./00024.jpg
./00151.jpg

./00095.jpg
./00618.jpg

./01726.jpg
./03087.jpg

About Datasets

Hello, I am very glad that you have made contributions to this field. May I ask where can you download your data set? It was not found in the file you uploaded.Thank you very much!

False detection.

have a lot of false detection, detecting people without helmet as people.

make all -j8 error

when i run the command line make all -j12 i found this error :

PROTOC src/caffe/proto/caffe.proto make: protoc: Command not found CXX src/caffe/layer_factory.cpp Makefile:632: recipe for target '.build_release/src/caffe/proto/caffe.pb.cc' failed make: *** [.build_release/src/caffe/proto/caffe.pb.cc] Error 127 make: *** Waiting for unfinished jobs.... CXX src/caffe/common.cpp src/caffe/common.cpp:1:10: fatal error: boost/thread.hpp: No such file or directory #include <boost/thread.hpp> ^~~~~~~~~~~~~~~~~~ compilation terminated. src/caffe/layer_factory.cpp:4:10: fatal error: boost/python.hpp: No such file or directory #include <boost/python.hpp> ^~~~~~~~~~~~~~~~~~ compilation terminated. Makefile:575: recipe for target '.build_release/src/caffe/common.o' failed make: *** [.build_release/src/caffe/common.o] Error 1 Makefile:575: recipe for target '.build_release/src/caffe/layer_factory.o' failed make: *** [.build_release/src/caffe/layer_factory.o] Error 1

yolov3-tiny score

I downloaded yolov3-tiny_final.weights and yolov3-tiny.cfg and compile model like this

helmet_net = cv2.dnn.readNetFromDarknet(
    'cfg/yolov3-tiny.cfg', 'model-weights/yolov3-tiny_final.weights')
helmet_net.setPreferableBackend(cv2.dnn.DNN_BACKEND_OPENCV)
helmet_net.setPreferableTarget(cv2.dnn.DNN_TARGET_CPU)
blob = cv2.dnn.blobFromImage(image, 1 / 255, (416, 416),
                                 [0, 0, 0], 1, crop=False)
helmet_net.setInput(blob)

# Runs the forward pass to get output of the output layers
outs = helmet_net.forward(get_outputs_names(helmet_net))

after this I map outs to get only items where confidences are >0.7 and get following result

result

What can be a problem of this behavior? Actually it didn't find real helmet...

the value of final loss

could you please share your value of final loss when you train the models like ssd or ssd-rpa, when I train it, the loss can't descent to a small value(the min loss: about 3.xxx).

About the training data

  你好,我前些天也用SSD算法识别安全帽,不过是基于tensorflow的,准确度大概70%,我想知道你是怎么制作训练样本的,不知道能不能发几个样本图片到我的邮箱?[email protected].顺便想问下你识别的准确度是多少啊?最后谢谢你分享到github上。

model weight file

good job, could you share the model weight files again? It seems that download link(both google driver and baidu) is not available

How many training images are enough

Your model is great!
How many training images are enough? such as to raise the mAP to 70%.
I have only a few images(100+) so my own model's mAP is lower than 50%.
I don't know if I need to append some training data?

MaskPooling layer is not available.

F1008 18:36:51.900141 31887 layer_factory.hpp:81] Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer type: MaskPooling (known types: AbsVal, Accuracy, AnnotatedData, ArgMax, BNLL, BatchNorm, BatchReindex, Bias, Concat, ContrastiveLoss, Convolution, Crop, Data, Deconvolution, DetectionEvaluate, DetectionOutput, Dropout, DummyData, ELU, Eltwise, Embed, EuclideanLoss, Exp, Filter, Flatten, HDF5Data, HDF5Output, HingeLoss, Im2col, ImageData, InfogainLoss, InnerProduct, Input, LRN, LSTM, LSTMUnit, Log, MVN, MemoryData, MultiBoxLoss, MultinomialLogisticLoss, Normalize, PReLU, Parameter, Permute, Pooling, Power, PriorBox, RNN, ReLU, Reduction, Reshape, SPP, Scale, Sigmoid, SigmoidCrossEntropyLoss, Silence, Slice, SmoothL1Loss, Softmax, SoftmaxWithLoss, Split, TanH, Threshold, Tile, VideoData, WindowData)

I built the caffe and trained Pelee-RPA for 120000 iterations. Now, I am trying to make inference and it gave me this error. Could anyone have an idea about this problem? - Thanks.

License

Hello @wujixiu, can you please add LICENSE file to your repository?

I would like to use some of the pretrained models (Squeezenet-SSD and SSD-RPA300) on your dataset but it is not clear what license they are available under. Currently the is no license information which technically means that your repo falls under "all rights reserved".

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.