Python：编写识别对象并对其进行计数的代码

Question

我正在编写一个python程序，使用colab计算和识别图像中的对象，当我在以下代码中遇到错误时，

drive_url = 'https://drive.google.com/uc?export=download&confirm=jH_3&id=' + DATASET_DRIVEID
file_name = DATASET_DRIVEID + '.zip'

urllib.request.urlretrieve(drive_url, file_name)
print('Download completed!')

os.makedirs(DATASET_DIR, exist_ok=True)
with zipfile.ZipFile(file_name, 'r') as zip_ref:
  zip_ref.extractall(DATASET_DIR)
os.remove(file_name)
print('Extract completed!')

文件下载已成功完成，但我认为提取部分中的错误显示为以下错误：

Download completed!
---------------------------------------------------------------------------
BadZipFile                                Traceback (most recent call last)
<ipython-input-5-6646b94bdb6f> in <module>()
      6 
      7 os.makedirs(DATASET_DIR, exist_ok=True)
----> 8 with zipfile.ZipFile(file_name, 'r') as zip_ref:
      9   zip_ref.extractall(DATASET_DIR)
     10 os.remove(file_name)

/usr/lib/python3.6/zipfile.py in __init__(self, file, mode, compression, allowZip64)
   1129         try:
   1130             if mode == 'r':
-> 1131                 self._RealGetContents()
   1132             elif mode in ('w', 'x'):
   1133                 # set the modified flag so central directory gets written

/usr/lib/python3.6/zipfile.py in _RealGetContents(self)
   1196             raise BadZipFile("File is not a zip file")
   1197         if not endrec:
-> 1198             raise BadZipFile("File is not a zip file")
   1199         if self.debug > 1:
   1200             print(endrec)

BadZipFile: File is not a zip file

有什么建议吗？

Answer 1

您的 Python 代码看起来正确，但 drive_url 的值除外。

错误消息表明名为file_name 的文件不是ZIP 文件。 urlretrieve 可能下载了其他内容，例如错误页面或不同格式的文件。可能您的drive_url 不正确（即不是您的浏览器使用的下载链接，而是网页链接），或者上传者同时从 Google Drive 中删除了该文件，或者您没有指定必要的 cookie下载。

使用您的网络浏览器下载drive_url，并查看下载的文件（例如，在Linux 上使用file 命令）。