图像分类处理小工具

最新推荐文章于 2024-04-22 10:07:56 发布

goodxin_ie

最新推荐文章于 2024-04-22 10:07:56 发布

阅读量847

点赞数 1

分类专栏： python 文章标签：分类人工智能机器学习

本文链接：https://blog.csdn.net/goodxin_ie/article/details/122469832

版权

功能：将图片移入对应文件夹

# -*- coding: utf-8 -*-
"""

This is a image labelTool script file.
parameter:  
      labelMap :   a dict like {0:'other',1:'cat',2:'dog',3:'car'}
      path :       image dir
This script will create some dirs in path. Each of those dirs will store pictures of one class in labelMap 
"""
from __future__ import print_function
import cv2
import os
import shutil

def createDir(path,labelMap):
    
    if(not isinstance(labelMap,dict)):
        print("labelMap must be a dict.....like {0:'cat',1:'dog'}")
        return False
    if(not os.path.exists(path)):
        print("{} is not exists!!!".format(path))
        return False
    for k,v in zip(labelMap.keys(),labelMap.values()):
        classPath = os.path.join(path,v)
        if(not os.path.exists(classPath)):
            os.mkdir(classPath)
    return True

        
def dealFile(fil