用于FashionAI关键点检测的卷积姿态机(CPM)和特征金字塔网词(FPN)的Pytorch实现
该代码引用了github上其他人的一些想法,并且基于的纸上。 最终排名是56,得分为6.38%。 对于大一新生来说,这对我来说还不错:)您可以将代码作为基准解决方案。
有一些切实可行的结果,如下所示:
使用ResNet-50的CPM + FPN输出示例
您可以按照以下步骤重现我的结果。
步骤1
相依性
Python 3.5版
脾气暴躁的
大熊猫
火炬
cv2
tqdm
Keras(TensorFlow后端)
资料准备
下载并将文件放入以下结构中:
步骤1:确保图像数据和注释位于折叠的“数据”中。数据目录结构如下:
DATA_DIR/
|-- test/
| |-- test.csv
| |-- Images/
| | |
2023-03-23 15:40:59
1.19MB
Python
1