N1955688260 2023-01-31 16:08 采纳率: 18.8%
浏览 22
已结题

关于#python#的问题:代码运行时报错

代码运行时报错,网上查找但无法解决,请求

import os
import os.path
def write_txt(content, filename, mode='w'):
    """保存txt数据
    :param content:需要保存的数据,type->list
    :param filename:文件名
    """
    with open(filename, mode) as f:
        for line in content:
            str_line = ""
            for col, data in enumerate(line):
                if not col == len(line) - 1:
                    # 以空格作为分隔符
                    str_line = str_line + str(data) + " "
                else:
                    # 每行最后一个数据用换行符“\n”
                    str_line = str_line + str(data) + "\n"
            f.write(str_line)
def get_files_list(dir):
    '''
    实现遍历dir目录下,所有文件(包含子文件夹的文件)
    :param dir:指定文件夹目录
    :return:包含所有文件的列表->list
    '''
    # parent:父目录, filenames:该目录下所有文件夹,filenames:该目录下的文件名
    files_list = []  # 写入文件的数据
    for parent, dirnames, filenames in os.walk(dir):
        for filename in filenames:
            print("parent is: " + parent)
            print("filename is: " + filename)
            print(os.path.join(parent, filename).replace('\\','/'))  # 输出rootdir路径下所有文件(包含子文件)信息
            curr_file = parent.split(os.sep)[-1]     # 获取正在遍历的文件夹名(也就是类名)
            # 根据class名确定labels
            if curr_file == "class1":
                labels = 0
            elif curr_file == "class1":
                labels = 1
            elif curr_file == "class1":
                labels = 2

            dir_path = parent.replace('\\', '/').split('/')[-2]   # train?val?test?

            curr_file = os.path.join(dir_path, curr_file)  # 相对路径

            files_list.append([os.path.join(curr_file, filename).replace('\\','/'), labels])     # 相对路径+label

    return files_list

if __name__ == '__main__':

    import pandas as pd

    #写入txt文件
    train_dir = 'D:/ningyupeng/pythonProject/data_cell_images/train'
    train_txt = 'D:/ningyupeng/pythonProject/data_cell_images/train.txt'
    train_data = get_files_list(train_dir)
    write_txt(train_data, train_txt, mode='w')

    val_dir = 'D:/ningyupeng/pythonProject/data_cell_images/val'
    val_txt = 'D:/ningyupeng/pythonProject/data_cell_images/val.txt'
    val_data = get_files_list(val_dir)
    write_txt(val_data, val_txt, mode='w')


```帮忙解决

错误信息:UnboundLocalError: local variable 'labels' referenced before assignment
  • 写回答

2条回答 默认 最新

  • 於黾 2023-01-31 16:16
    关注

    说明你的3个if一个都没有走,所以labels变量没有定义
    话说,你3个if的条件为什么都是curr_file == "class1"

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 2月8日
  • 已采纳回答 1月31日
  • 创建了问题 1月31日

悬赏问题

  • ¥15 VMBox虚拟机无法访问
  • ¥15 skd显示找不到头文件
  • ¥15 机器视觉中图片中长度与真实长度的关系
  • ¥15 R语言卸载之后无法重装,显示电脑存在下载某些较大二进制文件行为,怎么办
  • ¥15 java 的protected权限 ,问题在注释里
  • ¥15 这个是哪里有问题啊?
  • ¥15 关于#vue.js#的问题:修改用户信息功能图片无法回显,数据库中只存了一张图片(相关搜索:字符串)
  • ¥15 texstudio的问题,
  • ¥15 spaceclaim模型变灰色
  • ¥15 求一份华为esight平台V300R009C00SPC200这个型号的api接口文档