基于Faster R-CNN的安全帽目標(biāo)檢測
向AI轉(zhuǎn)型的程序員都關(guān)注了這個號???
機(jī)器學(xué)習(xí)AI算法工程?? 公眾號：datayx
訓(xùn)練模型前的準(zhǔn)備
A.數(shù)據(jù)準(zhǔn)備
數(shù)據(jù)的標(biāo)注仍然采用VOC格式的數(shù)據(jù)標(biāo)注形式，如果是其他的標(biāo)注形式比如COCO請自行實現(xiàn)相關(guān)代碼。將數(shù)據(jù)最終轉(zhuǎn)化為如下形式：
path_filename, x1, y1, x2, y2, class_name
格式說明
代碼及運行教程，數(shù)據(jù)集??獲取：
關(guān)注微信公眾號 datayx ?然后回復(fù)??安全帽? 即可獲取。
AI項目體驗地址 https://loveai.tech
可以運行如下代碼實現(xiàn)數(shù)據(jù)集的準(zhǔn)備工作:
python3 ./data/data_pro.py將在./data文件夾下生成annotation.txt文件，這樣訓(xùn)練數(shù)據(jù)的準(zhǔn)備工作即完成。
B.配置文件準(zhǔn)備
根據(jù)自己的訓(xùn)練集和訓(xùn)練任務(wù)修改./keras_frcnn/config.py的配置文件,相關(guān)參數(shù)的解釋和配置如下：
訓(xùn)練模型
預(yù)訓(xùn)練模型的下載地址：https://github.com/fchollet/deep-learning-models
Shell下運行：
python3 train_frcnn.py --path="./data/annotation.txt" --network="vgg" --input_weight_path="./pre_train/vgg16_weights_tf_kernels_notop.h5"
windows下直接運行我們寫好的批處理文件：
run_train.bat模型預(yù)測
將需要測試的圖像和視頻拷貝到./new_test文件夾
A.單張圖像推斷
Shell下運行：
python3 test_frcnn.py --path="./new_test"
windows下直接運行我們寫好的批處理文件：
run_inference.bat
B.視頻推斷
Shell下運行：
python3 test_frcnn_video.py --path="./new_test/test_video.mp4"
windows下直接運行我們寫好的批處理文件：
test_video.bat效果展示
基于yolo5工地安全帽和禁入危險區(qū)域識別系統(tǒng),附數(shù)據(jù)集
基于PyTorch+YOLOv4的口罩佩戴檢測，附數(shù)據(jù)集
閱讀過本文的人還看了以下文章：
TensorFlow 2.0深度學(xué)習(xí)案例實戰(zhàn)
基于40萬表格數(shù)據(jù)集TableBank，用MaskRCNN做表格檢測
《基于深度學(xué)習(xí)的自然語言處理》中/英PDF
Deep Learning 中文版初版-周志華團(tuán)隊
【全套視頻課】最全的目標(biāo)檢測算法系列講解，通俗易懂！
《美團(tuán)機(jī)器學(xué)習(xí)實踐》_美團(tuán)算法團(tuán)隊.pdf
《深度學(xué)習(xí)入門：基于Python的理論與實現(xiàn)》高清中文PDF+源碼
python就業(yè)班學(xué)習(xí)視頻，從入門到實戰(zhàn)項目
2019最新《PyTorch自然語言處理》英、中文版PDF+源碼
《21個項目玩轉(zhuǎn)深度學(xué)習(xí)：基于TensorFlow的實踐詳解》完整版PDF+附書代碼
《深度學(xué)習(xí)之pytorch》pdf+附書源碼
PyTorch深度學(xué)習(xí)快速實戰(zhàn)入門《pytorch-handbook》
【下載】豆瓣評分8.1,《機(jī)器學(xué)習(xí)實戰(zhàn):基于Scikit-Learn和TensorFlow》
《Python數(shù)據(jù)分析與挖掘?qū)崙?zhàn)》PDF+完整源碼
汽車行業(yè)完整知識圖譜項目實戰(zhàn)視頻(全23課)
李沐大神開源《動手學(xué)深度學(xué)習(xí)》，加州伯克利深度學(xué)習(xí)（2019春）教材
筆記、代碼清晰易懂！李航《統(tǒng)計學(xué)習(xí)方法》最新資源全套！
《神經(jīng)網(wǎng)絡(luò)與深度學(xué)習(xí)》最新2018版中英PDF+源碼
將機(jī)器學(xué)習(xí)模型部署為REST API
FashionAI服裝屬性標(biāo)簽圖像識別Top1-5方案分享
重要開源！CNN-RNN-CTC 實現(xiàn)手寫漢字識別
同樣是機(jī)器學(xué)習(xí)算法工程師，你的面試為什么過不了？
前海征信大數(shù)據(jù)算法：風(fēng)險概率預(yù)測
【Keras】完整實現(xiàn)‘交通標(biāo)志’分類、‘票據(jù)’分類兩個項目，讓你掌握深度學(xué)習(xí)圖像分類
VGG16遷移學(xué)習(xí)，實現(xiàn)醫(yī)學(xué)圖像識別分類工程項目
特征工程(二) :文本數(shù)據(jù)的展開、過濾和分塊
如何利用全新的決策樹集成級聯(lián)結(jié)構(gòu)gcForest做特征工程并打分？
Machine Learning Yearning 中文翻譯稿
斯坦福CS230官方指南：CNN、RNN及使用技巧速查（打印收藏）
python+flask搭建CNN在線識別手寫中文網(wǎng)站
中科院Kaggle全球文本匹配競賽華人第1名團(tuán)隊-深度學(xué)習(xí)與特征工程
不斷更新資源
深度學(xué)習(xí)、機(jī)器學(xué)習(xí)、數(shù)據(jù)分析、python
?搜索公眾號添加：?datayx??
機(jī)大數(shù)據(jù)技術(shù)與機(jī)器學(xué)習(xí)工程
?搜索公眾號添加：?datanlp
長按圖片，識別二維碼