/Cosmetic-Object-Detection

Object detection about cosmetic used by Faster RCNN.

Primary LanguageJupyter NotebookMIT LicenseMIT

Object detection about cosmetic used by Faster RCNN.

Dataset and Model

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中导入、发布、部署。

Training

python train-resnet.py

Test

python test.py

Results

效果图:

ModelArts部署: