Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

【Hackathon4 No.168】Add gupnet implementation #333

Open
wants to merge 17 commits into
base: develop
Choose a base branch
from

Conversation

Xiaolong-RRL
Copy link

  1. 增加GUPNet在paddle3d中的实现:
|-paddle3d-gupnet(ROOT)
  |-configs
    |-gupnet_dla34_kitti.py      # 配置文件
  |-docs
  	|-gupnet
  	  |-README.md	         # 说明文档(未完成)
  |-paddle3d
    |-datasets
      |-kitti
        |-gupnet_kitti.py        # 数据集加载
        |-gupnet_kitti_utils.py  # 数据集加载辅助函数
    |-models
      |-detection
        |-gupnet
          |-gupnet.py            # 模型结构搭建
          |-gupnet_dla.py        # 特征提取网络结构
          |-gupnet_scheduler.py	 # 学习率和优化器策略
          |-gupnet_helper.py	 # 辅助函数
          |-gupnet_loss.py       # 损失
          |-gupnet_predicter.py  # 预测
  |-paddle3d_gupnet_train.py     # 训练
  |-paddle3d_gupnet_val.py       # 验证
  1. 单卡、8batchsize的条件下,训练140个epoch,最后一个checkpoint的测试结果为 3D_AP_R40@70%: 21.44 15.44 12.84,截图如下:
    image

@Xiaolong-RRL Xiaolong-RRL changed the title add gupnet implementation 【PaddlePaddle Hackathon4 No.168】Add gupnet implementation May 16, 2023
@Xiaolong-RRL Xiaolong-RRL changed the title 【PaddlePaddle Hackathon4 No.168】Add gupnet implementation 【Hackathon4 No.168】Add gupnet implementation May 16, 2023
paddle3d/models/detection/gupnet/gupnet_scheduler.py Outdated Show resolved Hide resolved
paddle3d/models/detection/gupnet/gupnet_scheduler.py Outdated Show resolved Hide resolved
paddle3d/models/detection/gupnet/gupnet_trainer.py Outdated Show resolved Hide resolved
paddle3d_gupnet_train.py Outdated Show resolved Hide resolved
paddle3d_gupnet_val.py Outdated Show resolved Hide resolved
configs/gupnet/gupnet_dla34_kitti.yml Show resolved Hide resolved
configs/gupnet/gupnet_dla34_kitti.yml Outdated Show resolved Hide resolved


@manager.DATASETS.add_component
class GUPKittiMonoDataset(KittiDetDataset):
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

是不是可以统一成KittiMonoDataset

self.size_3d.apply(weights_init_xavier)
self.heading.apply(weights_init_xavier)

def export_forward(self, samples):
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

需要补充

@wangna11BD
Copy link
Collaborator

需要补充加copyright、readme和TIPC

.gitignore Outdated Show resolved Hide resolved
@wangna11BD
Copy link
Collaborator

.gitignore文件不用删~用原版的就可以,不要改~

@shiyutang
Copy link
Collaborator

@Xiaolong-RRL 非常感谢贡献,请问是否还有跟进意愿呢?

@shiyutang
Copy link
Collaborator

@juncaipeng 需要辛苦check下是否可以放入下一期黑客松赛题

@shiyutang shiyutang added the contributor External Developers label Sep 1, 2023
@nepeplwu nepeplwu assigned wangna11BD and unassigned juncaipeng Feb 1, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
contributor External Developers
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

5 participants