最近太多人问我如何绘制PR曲线了,我又很少及时看到你们的消息,在这里跟大家道个歉,我直接把代码贴出来,你们看着改参数就好。
################ 导入相关包 #####################
import os
import sys
import numpy as np
import tensorflow as tf
import matplotlib.pyplot as plt
ROOT_DIR = os.path.abspath("../../")
sys.path.append(ROOT_DIR) # To find local version of the library
from mrcnn import utils
import mrcnn.model as modellib
from samples.hpv import hpv # 这里是我自己写的脚本 继承的参数 一般是nucleus继承过来的
############## 配置参数 ####
LOGS_DIR = os.path.join(ROOT_DIR, "logs")
DATASET_DIR = os.path.join(ROOT_DIR, "datasets/hpv") # 数据集
config = hpv.NucleusInferenceConfig()
DEVICE = "/cpu:0"
TEST_MODE = "inference"
def get_ax(rows=1, cols=1, size=16):
fig, ax = plt.subplots(rows, cols, figsize=(size * cols, size * rows))
fig.tight_layout()
return ax
def text_save(filename, data):#filename为写入CSV文件的路径,data为要写入数据列表.
file =