文件名称:deepwaldo:使用MaskRCNN解决哪里的waldo
文件大小:51.23MB
文件格式:ZIP
更新时间:2024-02-25 02:32:50
localization deep-learning tensorflow cnn image-recognition
德瓦尔多 使用求解Waldo。 请参阅以获取有关问题和解决方案的详细说明。 一些结果 数据 用于此项目的培训和验证数据集位于文件夹中。 对于自定义数据集,请使用工具进行注释。 跑步 您可以在文件中设置配置,也可以使用默认设置。 然后使用以下命令进行训练和预测。 训练 这会将在可可数据集上训练的Mask-RCNN模型下载到MODEL_DIR文件夹,并使用DATA_DIR文件夹中的数据训练模型。 python train.py 检测中 显示带有在图像中检测到的瓦尔多的弹出窗口。 python predict.py [MODEL PATH] [PATH TO IMAGE] # for exam
【文件预览】:
deepwaldo-master
----.gitmodules(89B)
----models()
--------.gitignore(1B)
----data()
--------train()
--------val()
----train.py(1KB)
----waldo_config.py(516B)
----predict.py(966B)
----Mask_RCNN()
----LICENSE(1KB)
----results()
--------11.png(8.14MB)
--------5.png(6.3MB)
----.gitignore(21B)
----README.md(1KB)
----data.py(2KB)