python - 如何在谷歌colab中使用yad2k(yolo)

Question

感谢您阅读我的问题。我使用谷歌colab。我需要 yad2k 库。也许这不支持pip。

有人成功安装yad2k吗？

ImportError Traceback(最近一次调用最后一次) 在 () 13 从keras.models导入load_model，模型 14 从 yolo_utils 导入 read_classes、read_anchors、generate_colors、preprocess_image、draw_boxes、scale_boxes---> 15 从 yad2k.models.keras_yolo 导入 yolo_head、yolo_boxes_to_corners、preprocess_true_boxes、yolo_loss、yolo_body 16 17 get_ipython().magic('matplotlib内联')

/content/yad2k/models/keras_yolo.py () 中 10 11 从 utils 导入 compose---> 12 从 .keras_darknet19 导入(DarknetConv2D、DarknetConv2D_BN_Leaky、darknet_body) 13 14 sys.path.append('..')

/content/yad2k/models/keras_darknet19.py () 中 9 从 keras.regularizers 导入 l2 10---> 11 from ..utils import compose 12 13 # 具有静态默认参数的 Convolution2D 的部分包装器。

导入错误:无法导入名称“compose”

Answer 1

在colab中使用以下命令

!git 克隆 https://github.com/allanzelener/yad2k.git

从 yad2k.yad2k.models.keras_yolo 导入 yolo_head、yolo_boxes_to_corners、preprocess_true_boxes、yolo_loss、yolo_body

所有文件都在yad2k>yad2k>....所以你需要输入两次，希望你的问题得到解决