pytorch学习一:AlexNet颜色分类
数据集百度AI下载的颜色块数据集,一共7种颜色,不过都是合在一起的通过将图片转化到HSV空间,利用阈值提取不同颜色的图片,存放到不同文件夹,制作分类数据集。根据表格数据,提取颜色代码import numpy as npimport collectionsimport osimport cv2# 定义字典存放颜色分量上下限# 例如:{颜色: [min分量, max分量]}# {'red': [array([160, 43, 46]), array([179, 255, 255])]
原创
2021-09-04 21:14:55 ·
1794 阅读 ·
2 评论