-
大小: 116KB文件類型: .rar金幣: 2下載: 1 次發(fā)布日期: 2021-06-13
- 語言: Python
- 標(biāo)簽: yolo??yolo3??python??tensorflow??
資源簡介
yolo3的實(shí)現(xiàn),平臺(tái)是Windows10、python3.6.8 tensorflow1.12
運(yùn)行yolo_video.py --camera 打開本地?cái)z像頭識(shí)別
運(yùn)行yolo_video.py --image 打開圖片識(shí)別
修改yolo.py中的代碼,可以打開視頻文件
缺少權(quán)重文件,可以自行下載weight文件用convert.py轉(zhuǎn)換

代碼片段和文件信息
import?json
from?collections?import?defaultdict
name_box_id?=?defaultdict(list)
id_name?=?dict()
f?=?open(
????“mscoco2017/annotations/instances_train2017.json“
????encoding=‘utf-8‘)
data?=?json.load(f)
annotations?=?data[‘a(chǎn)nnotations‘]
for?ant?in?annotations:
????id?=?ant[‘image_id‘]
????name?=?‘mscoco2017/train2017/%012d.jpg‘?%?id
????cat?=?ant[‘category_id‘]
????if?cat?>=?1?and?cat?<=?11:
????????cat?=?cat?-?1
????elif?cat?>=?13?and?cat?<=?25:
????????cat?=?cat?-?2
????elif?cat?>=?27?and?cat?<=?28:
????????cat?=?cat?-?3
????elif?cat?>=?31?and?cat?<=?44:
????????cat?=?cat?-?5
????elif?cat?>=?46?and?cat?<=?65:
????????cat?=?cat?-?6
????elif?cat?==?67:
????????cat?=?cat?-?7
????elif?cat?==?70:
????????cat?=?cat?-?9
????elif?cat?>=?72?and?cat?<=?82:
????????cat?=?cat?-?10
????elif?cat?>=?84?and?cat?<=?90:
????????cat?=?cat?-?11
????name_box_id[name].append([ant[‘bbox‘]?cat])
f?=?open(‘train.txt‘?‘w‘)
for?key?in?name_box_id.keys():
????f.write(key)
????box_infos?=?name_box_id[key]
????for?info?in?box_infos:
????????x_min?=?int(info[0][0])
????????y_min?=?int(info[0][1])
????????x_max?=?x_min?+?int(info[0][2])
????????y_max?=?y_min?+?int(info[0][3])
????????box_info?=?“?%d%d%d%d%d“?%?(
????????????x_min?y_min?x_max?y_max?int(info[1]))
????????f.write(box_info)
????f.write(‘\n‘)
f.close()
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????文件???????1201??2018-07-31?07:42??yolo3\.gitignore
?????文件???????1323??2018-07-31?07:42??yolo3\coco_annotation.py
?????文件??????10093??2018-07-31?07:42??yolo3\convert.py
?????文件???????5707??2018-07-31?07:42??yolo3\darknet53.cfg
?????文件?????127344??2018-07-31?07:42??yolo3\font\FiraMono-Medium.otf
?????文件???????4432??2018-07-31?07:42??yolo3\font\SIL?Open?Font?License.txt
?????文件???????3482??2018-07-31?07:42??yolo3\kmeans.py
?????文件???????1064??2018-07-31?07:42??yolo3\LICENSE
?????文件????????625??2018-07-31?07:42??yolo3\model_data\coco_classes.txt
?????文件?????????50??2018-07-31?07:42??yolo3\model_data\tiny_yolo_anchors.txt
?????文件????????135??2018-07-31?07:42??yolo3\model_data\voc_classes.txt
?????文件?????????76??2018-07-31?07:42??yolo3\model_data\yolo_anchors.txt
?????文件???????4077??2018-07-31?07:42??yolo3\README.md
?????文件???????8332??2018-07-31?07:42??yolo3\train.py
?????文件??????10686??2018-07-31?07:42??yolo3\train_bottleneck.py
?????文件???????1416??2018-07-31?07:42??yolo3\voc_annotation.py
?????文件???????8592??2019-02-14?17:41??yolo3\yolo.py
?????文件??????16491??2018-07-31?07:42??yolo3\yolo3\model.py
?????文件???????3843??2018-07-31?07:42??yolo3\yolo3\utils.py
?????文件??????????0??2018-07-31?07:42??yolo3\yolo3\__init__.py
?????文件??????12868??2019-02-14?10:28??yolo3\yolo3\__pycache__\model.cpython-36.pyc
?????文件???????3844??2019-02-14?10:28??yolo3\yolo3\__pycache__\utils.cpython-36.pyc
?????文件????????117??2019-02-14?10:28??yolo3\yolo3\__pycache__\__init__.cpython-36.pyc
?????文件???????8342??2018-07-31?07:42??yolo3\yolo3.cfg
?????文件???????1915??2018-07-31?07:42??yolo3\yolov3-tiny.cfg
?????文件???????2631??2019-02-19?14:14??yolo3\yolo_video.py
?????文件???????7137??2019-02-14?17:37??yolo3\__pycache__\yolo.cpython-36.pyc
?????目錄??????????0??2019-02-14?10:28??yolo3\yolo3\__pycache__
?????目錄??????????0??2019-02-14?10:12??yolo3\font
?????目錄??????????0??2019-02-19?14:21??yolo3\model_data
............此處省略6個(gè)文件信息
評論
共有 條評論