add get pretrained method

autoai-org · Jan 23, 2019 · f006b1c · f006b1c
1 parent 70dc716
commit f006b1c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/cli/repository.go b/cli/repository.go
@@ -195,3 +195,11 @@ func InitNewRepo(repoName string) {
 	bolierplateURL := "https://github.com/cvmodel/bolierplate.git"
 	CloneFromGit(bolierplateURL, repoName)
 }
+
+func GetPretrained(vendorName string, packageName string) []os.FileInfo {
+	config := readConfig()
+	localFolder := filepath.Join(config.Local.LocalFolder, vendorName, packageName)
+	localPretrainedFolder := filepath.Join(localFolder, "pretrained")
+	files, _ := ioutil.ReadDir(localPretrainedFolder)
+	return files
+}