复制文件
#copy the file "test.py" in now path to the child path mytest/
if not os.path.exists("mytest"):
os.makedirs("mytest")
copy2("test.py","mytest/")
获取文件名
url="home/cx/test.txt"
filename = url.split('/')[-1]
print filename
利用logging输出
logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s',
level=logging.DEBUG,
stream=sys.stdout)
logging.info("Download URL: {}".format("xxxx"))
logging.debug('This is debug message')
logging.info('This is info message')
logging.warning('This is warning message')
def get_pathes():
if 'TV_DIR_DATA' in os.environ:
data_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA'])
else:
data_dir = "DATA"
if 'TV_DIR_RUNS' in os.environ:
run_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA'])
else:
run_dir = "RUNS"
return data_dir, run_dir
data_dir, run_dir = get_pathes()
print os.path.join(data_dir,"cx") #if data_dir=home,then print home/cx
print run_dir,data_dir
import os
import logging
import sys
from shutil import copy2
源代码(总)
"""Download data relevant to train the KittiSeg model."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import logging
import sys
import os
import subprocess
import zipfile
from six.moves import urllib
from shutil import copy2
import argparse
logging.basicConfig(format='%(asctime)s %(levelname)s %(message)s',
level=logging.INFO,
stream=sys.stdout)
sys.path.insert(1, 'incl')
# Please set kitti_data_url to the download link for the Kitti DATA.
#
# You can obtain by going to this website:
# http://www.cvlibs.net/download.php?file=data_road.zip
#
# Replace 'http://kitti.is.tue.mpg.de/kitti/?????????.???' by the
# correct URL.
vgg_url = 'https://dl.dropboxusercontent.com/u/50333326/vgg16.npy'
def get_pathes():
"""
Get location of `data_dir` and `run_dir'.
Defaut is ./DATA and ./RUNS.
Alternativly they can be set by the environoment variabels
'TV_DIR_DATA' and 'TV_DIR_RUNS'.
"""
if 'TV_DIR_DATA' in os.environ:
data_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA'])
else:
data_dir = "DATA"
if 'TV_DIR_RUNS' in os.environ:
run_dir = os.path.join(['hypes'], os.environ['TV_DIR_DATA'])
else:
run_dir = "RUNS"
return data_dir, run_dir
def download(url, dest_directory):
filename = url.split('/')[-1]
filepath = os.path.join(dest_directory, filename)
logging.info("Download URL: {}".format(url))
logging.info("Download DIR: {}".format(dest_directory))
def _progress(count, block_size, total_size):
prog = float(count * block_size) / float(total_size) * 100.0
sys.stdout.write('\r>> Downloading %s %.1f%%' %
(filename, prog))
sys.stdout.flush()
filepath, _ = urllib.request.urlretrieve(url, filepath,
reporthook=_progress)
print()
return filepath
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--kitti_url', default='', type=str)
args = parser.parse_args()
kitti_data_url = args.kitti_url
data_dir, run_dir = get_pathes()
vgg_weights = os.path.join(data_dir, 'vgg16.npy')
if not os.path.exists(data_dir):
os.makedirs(data_dir)
# Download VGG DATA
if not os.path.exists(vgg_weights):
download_command = "wget {} -P {}".format(vgg_url, data_dir)
logging.info("Downloading VGG weights.")
download(vgg_url, data_dir)
else:
logging.warning("File: {} exists.".format(vgg_weights))
logging.warning("Please delete to redownload VGG weights.")
data_road_zip = os.path.join(data_dir, 'data_road.zip')
# Download KITTI DATA
if not os.path.exists(data_road_zip):
if kitti_data_url == '':
logging.error("Data URL for Kitti Data not provided.")
url = "http://www.cvlibs.net/download.php?file=data_road.zip"
logging.error("Please visit: {}".format(url))
logging.error("and request Kitti Download link.")
logging.error("Rerun scipt using"
"'python download_data.py' --kitti_url [url]")
exit(1)
if not kitti_data_url[-19:] == 'kitti/data_road.zip':
logging.error("Wrong url.")
url = "http://www.cvlibs.net/download.php?file=data_road.zip"
logging.error("Please visit: {}".format(url))
logging.error("and request Kitti Download link.")
logging.error("Rerun scipt using"
"'python download_data.py' --kitti_url [url]")
exit(1)
else:
logging.info("Downloading Kitti Road Data.")
download(kitti_data_url, data_dir)
# Extract and prepare KITTI DATA
logging.info("Extracting kitti_road data.")
zipfile.ZipFile(data_road_zip, 'r').extractall(data_dir)
kitti_road_dir = os.path.join(data_dir, 'data_road/')
logging.info("Preparing kitti_road data.")
train_txt = "data/train3.txt"
val_txt = "data/val3.txt"
testing_txt = "data/testing.txt"
copy2(train_txt, kitti_road_dir)
copy2(val_txt, kitti_road_dir)
copy2(testing_txt, kitti_road_dir)
logging.info("All data have been downloaded successful.")
if __name__ == '__main__':
main()