当前位置:网站首页>开源|商品识别推荐系统
开源|商品识别推荐系统
2022-07-31 06:48:00 【思考实践】
几个月前做的项目,整理一下,比较简单,就是根据数据集进行图片分类(使用resnet50),再通过识别结果匹配推荐数据库,进行推荐。
前端:html、js
后端:python-flask框架、pytorch框架
项目结构:
dataset_dir文件夹:数据集放进这个文件夹,不同种类图片放进对应种类文件夹,再通过
models文件夹:存放训练好的模型
static文件夹:存放前端网页显示的需要的图片
templates文件夹:前端网页模板文件
inference.py 测试推理效果的文件
label.py
from torchvision.datasets import ImageFolder
dataset=ImageFolder("C:/Users/forwhat/Desktop/recommend/cloth/train") #获取路径,返回的是所有图的data、label
print(dataset.class_to_idx) #查看类别名,及对应的标签。
print(dataset.imgs) #查看路径里所有的图片,及对应的标签
label_level2.py
under_jeans =['https://detail.tmall.com/item.htm?id=636464038643','https://detail.tmall.com/item.htm?id=629142114736','https://item.taobao.com/item.htm?id=633482107680','https://item.taobao.com/item.htm?id=653135699650'
,'https://detail.tmall.com/item.htm?id=626092831491']
under_skirt = ['https://item.taobao.com/item.htm?id=667268983604','https://detail.tmall.com/item.htm?id=668100149427','https://detail.tmall.com/item.htm?id=586698989166','https://detail.tmall.com/item.htm?id=635332780625'
,'https://detail.tmall.com/item.htm?id=637917736279']
under_sporty = ['https://detail.tmall.com/item.htm?id=613077657236','https://detail.tmall.com/item.htm?id=580354266231','https://detail.tmall.com/item.htm?id=580354266231','https://detail.tmall.com/item.htm?id=653363447356','https://item.taobao.com/item.htm?id=579123847078'
,'https://detail.tmall.com/item.htm?id=42730275588']
under_suit = ['https://detail.tmall.com/item.htm?id=618731089852','https://detail.tmall.com/item.htm?id=618596933879','https://detail.tmall.com/item.htm?id=636256949983','https://detail.tmall.com/item.htm?id=636363512038'
,'https://item.taobao.com/item.htm?id=665038582357']
upper_casual = ['https://detail.tmall.com/item.htm?id=666855488772','https://detail.tmall.com/item.htm?id=667962767730','https://detail.tmall.com/item.htm?id=610974657070','https://detail.tmall.com/item.htm?id=645567560209'
,'https://detail.tmall.com/item.htm?id=644415535799']
upper_coat = ['https://item.taobao.com/item.htm?id=552053460493','https://item.taobao.com/item.htm?id=598306760200','https://detail.tmall.com/item.htm?id=665519667811','https://detail.tmall.com/item.htm?id=585993479022'
,'https://item.taobao.com/item.htm?id=649233831246']
upper_hoodie = ['https://detail.tmall.com/item.htm?id=666165898023','https://item.taobao.com/item.htm?id=652872409153','https://item.taobao.com/item.htm?id=628894958150','https://detail.tmall.com/item.htm?id=633528937891'
,'https://item.taobao.com/item.htm?id=656080489764']
upper_sporty = ['https://item.taobao.com/item.htm?id=628559051772','https://detail.tmall.com/item.htm?id=668803933152','https://detail.tmall.com/item.htm?id=655672026380','https://detail.tmall.com/item.htm?id=631357306618'
,'https://detail.tmall.com/item.htm?id=601009441405']
upper_suit = ['https://detail.tmall.com/item.htm?id=666236552324','https://detail.tmall.com/item.htm?id=668803933152','https://item.taobao.com/item.htm?id=625828294674','https://detail.tmall.com/item.htm?id=649272374842','https://detail.tmall.com/item.htm?id=666063758350'
,'https://item.taobao.com/item.htm?id=668250478482']
whole_dress = ['https://detail.tmall.com/item.htm?id=598268797504','https://item.taobao.com/item.htm?id=666364831194','https://detail.tmall.com/item.htm?id=667095587171','https://detail.tmall.com/item.htm?id=642013218199'
,'https://detail.tmall.com/item.htm?id=669133765583']
split_dataset.py脚本对数据集进行训练集、验证集、测试集的划分。
train.py脚本对数据集进行训练
数据集
符合格式就可以,参考链接一
效果展示
点击一个链接看看
开源地址
基于神经网络的简单推荐系统: 几个月前做的项目,整理一下,比较简单,就是根据数据集进行图片分类(使用resnet50),再通过识别结果匹配推荐数据库,进行推荐。 (gitee.com)
参考资料
(195条消息) 【深度学习】使用python划分数据集为训练集和验证集和测试集并放在不同的文件夹_我系渣渣的博客-CSDN博客_python划分数据集为训练集和测试集
边栏推荐
猜你喜欢
把 VS Code 当游戏机
Embedded system driver primary [2] - _ parameters and dependencies under the kernel module
Automatic translation software - batch batch automatic translation software recommendation
[PSQL] SQL Basic Course Reading Notes (Chapter1-4)
双倍数据速率同步动态随机存储器(Double Data Rate Synchronous Dynamic Random Access Memory, DDR SDRAM)- 逻辑描述部分
【愚公系列】2022年07月 Go教学课程 022-Go容器之字典
【微服务】(十六)—— 分布式事务Seata
【科普向】5G核心网架构和关键技术
MySQL安装到最后一步 write configuration file 失败 怎么办?及后安装步骤
Yu Mr Series 】 【 2022 July 022 - Go Go teaching course of container in the dictionary
随机推荐
2022.07.15_Daily Question
2022.07.15_Daily Question
【Go报错】go go.mod file not found in current directory or any parent directory 错误解决
科普 | “大姨太”ETH 和 “小姨太”ETC的爱恨情仇
The Perfect Guide|How to use ODBC for Agentless Oracle Database Monitoring?
CNN--Introduction to each layer
mysql的建表语句_三种常用的MySQL建表语句
2022.07.24_Daily Question
强化学习科研知识必备(数据库、期刊、会议、牛人)
[Interview: Concurrency 38: Multithreading: Thread Pool] Basic concepts of the ThreadPoolExecutor class
Pygame Surface对象
ros little turtle drawing
2022.07.20_每日一题
2022.7.29 Array
我开发了一个利用 Bun 执行 .ts / .js 文件的 VS Code 插件
LeetCode:952. 按公因数计算最大组件大小【欧拉筛 + 并查集】
XSS靶场prompt.ml过关详解
2022.07.20_Daily Question
Client navicat installation tutorial
【Go语言入门】一文搞懂Go语言的最新依赖管理:go mod的使用