Merge remote-tracking branch 'origin/master' into cnet
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
import os
|
||||
import urllib.request as dowReq
|
||||
import zipfile
|
||||
|
||||
val = input("Enter BDD or COCO :")
|
||||
if(val == "COCO"):
|
||||
url = "https://cloud.hipert.unimore.it/s/LNxBDk4wzqXPL8c/download"
|
||||
lib = "..\demo\COCO_val2017"
|
||||
lib_zip = "COCO_val2017.zip"
|
||||
elif(val == "BDD"):
|
||||
url = "https://cloud.hipert.unimore.it/s/bikqk3FzCq2tg4D/download"
|
||||
lib = "..\demo\BDD100k_val"
|
||||
lib_zip = "BDD100k_val.zip"
|
||||
|
||||
dowReq.urlretrieve(url,lib_zip)
|
||||
|
||||
with zipfile.ZipFile(lib_zip,'r') as zip_ref:
|
||||
zip_ref.extractall(lib)
|
||||
|
||||
labelFolder = lib + "\labels"
|
||||
imageFolder = lib + "\images"
|
||||
|
||||
file1 = open(".\\..\\demo\\all_labels.txt","a")
|
||||
path1 = os.path.realpath(labelFolder)
|
||||
for file in os.listdir(labelFolder):
|
||||
valTemp = path1 + "\\" + file
|
||||
valTemp = valTemp + '\n'
|
||||
file1.write(valTemp)
|
||||
file1.close()
|
||||
|
||||
file2 = open(".\\..\\demo\\all_images.txt","a")
|
||||
path2 = os.path.realpath(imageFolder)
|
||||
for file in os.listdir(imageFolder):
|
||||
pathtemp = path2 + "\\" + file
|
||||
pathtemp = pathtemp + '\n'
|
||||
file2.write(pathtemp)
|
||||
file2.close()
|
||||
|
||||
print("Completed")
|
||||
+22
-21
@@ -69,30 +69,31 @@ do
|
||||
echo -e "${ORANGE}Batch $TKDNN_BATCHSIZE ${NC}"
|
||||
|
||||
test_net mnist
|
||||
./test_imuodom &>> $out_file
|
||||
print_output $? imuodom
|
||||
# ./test_imuodom &>> $out_file
|
||||
# print_output $? imuodom
|
||||
|
||||
# test_net yolo4
|
||||
# test_net yolo4x
|
||||
# test_net yolo4_berkeley
|
||||
# test_net yolo4tiny
|
||||
# test_net yolo3
|
||||
# test_net yolo3_berkeley
|
||||
# test_net yolo3_coco4
|
||||
# test_net yolo3_flir
|
||||
# test_net yolo3_512
|
||||
# test_net yolo3tiny
|
||||
# test_net yolo3tiny_512
|
||||
# test_net yolo2
|
||||
# test_net yolo2_voc
|
||||
test_net yolo4
|
||||
test_net yolo4-csp
|
||||
test_net yolo4x
|
||||
test_net yolo4_berkeley
|
||||
test_net yolo4tiny
|
||||
test_net yolo3
|
||||
test_net yolo3_berkeley
|
||||
test_net yolo3_coco4
|
||||
test_net yolo3_flir
|
||||
test_net yolo3_512
|
||||
test_net yolo3tiny
|
||||
test_net yolo3tiny_512
|
||||
test_net yolo2
|
||||
test_net yolo2_voc
|
||||
#test_net yolo2tiny
|
||||
# test_net csresnext50-panet-spp
|
||||
test_net csresnext50-panet-spp
|
||||
#test_net csresnext50-panet-spp_berkeley
|
||||
# test_net resnet101_cnet
|
||||
# test_net dla34_cnet
|
||||
# test_net mobilenetv2ssd
|
||||
# test_net mobilenetv2ssd512
|
||||
# test_net bdd-mobilenetv2ssd
|
||||
test_net resnet101_cnet
|
||||
test_net dla34_cnet
|
||||
test_net mobilenetv2ssd
|
||||
test_net mobilenetv2ssd512
|
||||
test_net bdd-mobilenetv2ssd
|
||||
test_net dla34_cnet3d_track
|
||||
done
|
||||
|
||||
|
||||
Reference in New Issue
Block a user