使用说明：
按照requirements.txt安装相应的环境包
ubuntu20.04+python3.8+cuda3070显卡加速
训练说明：
1.将数据集(包含图片和标注的数据集)放入data文件夹
2.配置好相关环境
3.使用下面命令运行训练模型：
python main.py -m dab_detr --output_dir logs/DABDETR/R50 --batch_size 1 --epochs 50 --lr_drop 40 --coco_path ./data
4. .\transformer\logs\DABDETR\R50w存储训练后的模型
info.txt 存储训练相关信息，log.txt存储训练过程
figure 存储网络架构图
测试说明：
运行python predict.py
测试结果图片存储在result文件夹下。
为了便于发送：
我把数据集中的图片文件夹删除了。
基于transformer的车辆行人识别
版权申诉
5星 · 超过95%的资源 130 浏览量
2022-06-05
14:41:54
上传
评论 7
收藏 954.89MB RAR 举报 
基于transformer的车辆行人识别 （137个子文件） 
ms_deform_attn_cpu.cpp 1KB vision.cpp 799B ms_deform_attn_cuda.cu 7KB ms_deform_im2col_cuda.cuh 53KB MultiScaleDeformableAttention-1.0-py3.8-linux-x86_64.egg 2.45MB ms_deform_attn.h 2KB ms_deform_attn_cuda.h 1KB ms_deform_attn_cpu.h 1KB instances_train2017.json 4.64MB instances_val2017.json 1.03MB image_info_test-dev2017.json 1006KB config.json 2KB config.json 2KB readme.md 708B vision.o 12.86MB ms_deform_attn_cpu.o 1.45MB ms_deform_attn_cuda.o 883KB PKG-INFO 311B
2.png 1.2MB 1.png 1.12MB results.png 248KB model.png 201KB arch_new2.png 153KB arch_new.png 153KB arch.png 71KB 数据集目录.png 15KB
comparison_raleted_works_raw.pptx 210KB checkpoint.pth 541.65MB checkpoint.pth 498.92MB 000.pth 835KB latest.pth 834KB 200.pth 634KB 150.pth 633KB latest.pth 632KB eval.pth 632KB 250.pth 632KB 100.pth 631KB 050.pth 622KB 000.pth 581KB swin_transformer.py 26KB dab_deformable_detr.py 25KB DABDETR.py 23KB deformable_transformer.py 22KB attention.py 20KB transformer.py 20KB main.py 19KB misc.py 19KB segmentation.py 16KB segmentation.py 16KB slconfig.py 14KB utils.py 13KB engine.py 11KB coco.py 9KB coco_eval.py 9KB visualizer.py 9KB transforms.py 8KB sltransform.py 8KB ms_deform_attn.py 6KB ms_deform_attn.py 6KB position_encoding.py 6KB backbone.py 6KB backbone.py 5KB slio.py 5KB random_crop.py 5KB matcher.py 5KB matcher.py 5KB plot_utils.py 5KB test.py 4KB position_encoding.py 4KB box_ops.py 4KB box_loss.py 4KB coco_panoptic.py 4KB logger.py 3KB get_param_dicts.py 3KB ms_deform_attn_func.py 3KB ms_deform_attn_func.py 3KB vis_utils.py 3KB setup.py 3KB predict.py 2KB time_counter.py 2KB panoptic_eval.py 1KB hostlist.py 1KB __init__.py 897B __init__.py 816B __init__.py 792B __init__.py 598B __init__.py 598B __init__.py 584B __init__.py 584B __init__.py 379B __init__.py 71B swin_transformer.cpython-38.pyc 20KB dab_deformable_detr.cpython-38.pyc 20KB DABDETR.cpython-38.pyc 19KB misc.cpython-38.pyc 18KB deformable_transformer.cpython-38.pyc 16KB attention.cpython-38.pyc 15KB utils.cpython-38.pyc 15KB segmentation.cpython-38.pyc 13KB slconfig.cpython-38.pyc 13KB共 137 条
- 1
- 2
图灵追慕者
- 粉丝: 2638
- 资源: 102
最新资源
- lab7_02.c
- 基于 SpringCloud 和 Vue3 的OA系统
- 软考高级项目管理师-项目采购管理思维导图
- WordsVector (1).ipynb
- yolov7 车牌检测 车牌识别 中文车牌识别 检测 支持双层车牌 支持12种中文车牌
- 20240429_112025.m4a
- "麦嘟学编程"似乎是一个与编程学习相关的品牌或社区名称，它可能是一个在线教育平台、博客、论坛或社交媒体群组等，旨在帮助人们学习编
- OpenCV（Open Source Computer Vision Library）是一个开源的计算机视觉和机器学习软件库，由
- Redis入门基础篇+源码（springboot、maven）
- ChatGPT-4是由OpenAI开发的人工智能模型，是GPT（Generative Pre-trained Transform
资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！
点击此处反馈
- 1
- 2
前往页