- 本实验代码的IDE使用的是VsCode,右键通过VsCode打开
Code
文件夹即可 - 为了方便老师运行检查,本项目已经跑完
train.py
生成了50个权值文件,选出了其中一个作为frcnn.py
的model_path
的参数(使用的是ResNet,训练的损失函数值如下图所示),运行结果具体可见code\logs\loss_2021_12_02_13_46_18
文件夹中的文件 - 如果要完整运行整个项目,需要先运行
voc_annotation.py
,再运行train.py
,修改frcnn.py
的model_path
的参数为log
文件夹中的权值文件,运行predict.py
,输入图片进行预测,最后运行get_map.py
计算mAP值
-
Notifications
You must be signed in to change notification settings - Fork 0
ytWu1314/Object-detection-based-on-Faster-R-CNN
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
SCNU Machine learning projects
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published