KITTI數(shù)據(jù)集是自動(dòng)駕駛領(lǐng)域的經(jīng)典數(shù)據(jù)集之一，raw data大小有上百G，官方在raw data界面提供了一個(gè)shell腳本raw dataset download script來幫助下載，該shell腳本會(huì)自動(dòng)下載所有raw data并解壓到指定文件夾。
由于raw data過大，斷斷續(xù)續(xù)下了好幾天，后來在用數(shù)據(jù)的時(shí)候發(fā)現(xiàn)有的圖像出現(xiàn)損壞，有的文件夾下沒有文件，遂想重新下載一下未解壓的raw data，把官方的shell腳本改成了python腳本，代碼如下：

import wget
import os

files_name = ["2011_09_26_calib.zip",
"2011_09_26_drive_0001",
"2011_09_26_drive_0002",
"2011_09_26_drive_0005",
"2011_09_26_drive_0009",
"2011_09_26_drive_0011",
"2011_09_26_drive_0013",
"2011_09_26_drive_0014",
"2011_09_26_drive_0015",
"2011_09_26_drive_0017",
"2011_09_26_drive_0018",
"2011_09_26_drive_0019",
"2011_09_26_drive_0020",
"2011_09_26_drive_0022",
"2011_09_26_drive_0023",
"2011_09_26_drive_0027",
"2011_09_26_drive_0028",
"2011_09_26_drive_0029",
"2011_09_26_drive_0032",
"2011_09_26_drive_0035",
"2011_09_26_drive_0036",
"2011_09_26_drive_0039",
"2011_09_26_drive_0046",
"2011_09_26_drive_0048",
"2011_09_26_drive_0051",
"2011_09_26_drive_0052",
"2011_09_26_drive_0056",
"2011_09_26_drive_0057",
"2011_09_26_drive_0059",
"2011_09_26_drive_0060",
"2011_09_26_drive_0061",
"2011_09_26_drive_0064",
"2011_09_26_drive_0070",
"2011_09_26_drive_0079",
"2011_09_26_drive_0084",
"2011_09_26_drive_0086",
"2011_09_26_drive_0087",
"2011_09_26_drive_0091",
"2011_09_26_drive_0093",
"2011_09_26_drive_0095",
"2011_09_26_drive_0096",
"2011_09_26_drive_0101",
"2011_09_26_drive_0104",
"2011_09_26_drive_0106",
"2011_09_26_drive_0113",
"2011_09_26_drive_0117",
"2011_09_26_drive_0119",
"2011_09_28_calib.zip",
"2011_09_28_drive_0001",
"2011_09_28_drive_0002",
"2011_09_28_drive_0016",
"2011_09_28_drive_0021",
"2011_09_28_drive_0034",
"2011_09_28_drive_0035",
"2011_09_28_drive_0037",
"2011_09_28_drive_0038",
"2011_09_28_drive_0039",
"2011_09_28_drive_0043",
"2011_09_28_drive_0045",
"2011_09_28_drive_0047",
"2011_09_28_drive_0053",
"2011_09_28_drive_0054",
"2011_09_28_drive_0057",
"2011_09_28_drive_0065",
"2011_09_28_drive_0066",
"2011_09_28_drive_0068",
"2011_09_28_drive_0070",
"2011_09_28_drive_0071",
"2011_09_28_drive_0075",
"2011_09_28_drive_0077",
"2011_09_28_drive_0078",
"2011_09_28_drive_0080",
"2011_09_28_drive_0082",
"2011_09_28_drive_0086",
"2011_09_28_drive_0087",
"2011_09_28_drive_0089",
"2011_09_28_drive_0090",
"2011_09_28_drive_0094",
"2011_09_28_drive_0095",
"2011_09_28_drive_0096",
"2011_09_28_drive_0098",
"2011_09_28_drive_0100",
"2011_09_28_drive_0102",
"2011_09_28_drive_0103",
"2011_09_28_drive_0104",
"2011_09_28_drive_0106",
"2011_09_28_drive_0108",
"2011_09_28_drive_0110",
"2011_09_28_drive_0113",
"2011_09_28_drive_0117",
"2011_09_28_drive_0119",
"2011_09_28_drive_0121",
"2011_09_28_drive_0122",
"2011_09_28_drive_0125",
"2011_09_28_drive_0126",
"2011_09_28_drive_0128",
"2011_09_28_drive_0132",
"2011_09_28_drive_0134",
"2011_09_28_drive_0135",
"2011_09_28_drive_0136",
"2011_09_28_drive_0138",
"2011_09_28_drive_0141",
"2011_09_28_drive_0143",
"2011_09_28_drive_0145",
"2011_09_28_drive_0146",
"2011_09_28_drive_0149",
"2011_09_28_drive_0153",
"2011_09_28_drive_0154",
"2011_09_28_drive_0155",
"2011_09_28_drive_0156",
"2011_09_28_drive_0160",
"2011_09_28_drive_0161",
"2011_09_28_drive_0162",
"2011_09_28_drive_0165",
"2011_09_28_drive_0166",
"2011_09_28_drive_0167",
"2011_09_28_drive_0168",
"2011_09_28_drive_0171",
"2011_09_28_drive_0174",
"2011_09_28_drive_0177",
"2011_09_28_drive_0179",
"2011_09_28_drive_0183",
"2011_09_28_drive_0184",
"2011_09_28_drive_0185",
"2011_09_28_drive_0186",
"2011_09_28_drive_0187",
"2011_09_28_drive_0191",
"2011_09_28_drive_0192",
"2011_09_28_drive_0195",
"2011_09_28_drive_0198",
"2011_09_28_drive_0199",
"2011_09_28_drive_0201",
"2011_09_28_drive_0204",
"2011_09_28_drive_0205",
"2011_09_28_drive_0208",
"2011_09_28_drive_0209",
"2011_09_28_drive_0214",
"2011_09_28_drive_0216",
"2011_09_28_drive_0220",
"2011_09_28_drive_0222",
"2011_09_28_drive_0225",
"2011_09_29_calib.zip",
"2011_09_29_drive_0004",
"2011_09_29_drive_0026",
"2011_09_29_drive_0071",
"2011_09_29_drive_0108",
"2011_09_30_calib.zip",
"2011_09_30_drive_0016",
"2011_09_30_drive_0018",
"2011_09_30_drive_0020",
"2011_09_30_drive_0027",
"2011_09_30_drive_0028",
"2011_09_30_drive_0033",
"2011_09_30_drive_0034",
"2011_09_30_drive_0072",
"2011_10_03_calib.zip",
"2011_10_03_drive_0027",
"2011_10_03_drive_0034",
"2011_10_03_drive_0042",
"2011_10_03_drive_0047",
"2011_10_03_drive_0058"]

for file_name in files_name:
    if file_name.endswith('.zip'):
        short_name = file_name
        full_name = file_name
    else:
        short_name = file_name + '_sync.zip'
        full_name = file_name + '/' + file_name + '_sync.zip'

    url = 'https://s3.eu-central-1.amazonaws.com/avg-kitti/raw_data/' + full_name
    print("Downloading: " + short_name)
    print('url: ' + url)
    wget.download(url, out=short_name)

總結(jié)

以上是生活随笔為你收集整理的下载KITTI数据集raw data的Python脚本的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。