Dataset:https://github.com/xuyanbo03/Cosmetic-Object-Detection/tree/master/data/VOCdevkit2007/VOC2007
Model: 百度云
提取码:vjfw
该模型使用Faster RCNN算法,并加载预训练权重 resnet_v1_50.ckpt 进行训练,数据集为VOC2007格式,最终实现效果为识别护肤品(cosmetic)位置。
该model文件可在ModelArts中导入、发布、部署。
python train-resnet.py
python test.py
效果图:
ModelArts部署: